std1.0.0-beta-7 SafeC docs

std :: ml::tensor

Trait std::RocmTensor

#include <std/ml/tensor.h> source

trait RocmTensor : DeviceTensor {
    unsigned long device_buffer() const;
}

One trait per GPU backend, each inheriting DeviceTensor: code written for one backend's buffers (a hand-written kernel, a vendor library call) asks for its tensor. device_buffer() is the values' buffer on that backend -- a hipDeviceptr, a CUdeviceptr, an MTLBuffer's GPU address, a Vulkan buffer device address -- and 0 while the tensor is elsewhere: check device() first, or place() it.

Required methods

§unsigned long device_buffer() constsource