`tensor.tensor.Tensor`, `max.tensor.Tensor` or `max.driver.Tensor`?
Are we meant to use
tensor.tensor.Tensor
, max.tensor.Tensor
or max.driver.Tensor
? The first one might be a re-export of the second, but max.driver.Tensor
is the only one which seems to handle multi-device, however most of the max APIs don't seem to work on it despite it seeming to be a better match for ManagedTensorSlice
. However, I don't see a way to move data off of the GPU with the max.tensor.Tensor
/tensor.tensor.Tensor
variants, and I get segfaults in what looks like a memcpy from a GPU pointer to a CPU pointer.0 Replies