MCPcopy Create free account
hub / github.com/MrNeRF/LichtFeld-Studio / is_cuda

Method is_cuda

src/python/lfs/py_tensor.cpp:217–219  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

215 }
216
217 bool PyTensor::is_cuda() const {
218 return tensor_.device() == Device::CUDA;
219 }
220
221 size_t PyTensor::size(int dim) const {
222 const int resolved = dim < 0 ? static_cast<int>(tensor_.shape().rank()) + dim : dim;

Callers 7

TEST_FFunction · 0.80
copy_to_torchMethod · 0.80
from_torchFunction · 0.80
TEST_FFunction · 0.80
TESTFunction · 0.80

Calls 1

deviceMethod · 0.45

Tested by 6

TEST_FFunction · 0.64
copy_to_torchMethod · 0.64
TEST_FFunction · 0.64
TESTFunction · 0.64