`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
No replies yetBe the first to reply to this messageJoin

Did you find this page helpful?