MCPcopy Create free account
hub / github.com/UbiquitousLearning/mllm / cuda

Method cuda

mllm/core/Tensor.cpp:457–460  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

455}
456
457Tensor Tensor::cuda() {
458 if (kCUDA == impl_->device()) { return *this; }
459 return to(kCUDA);
460}
461
462Tensor Tensor::equal(Tensor v) {
463 return Context::instance().buildOpAndSubmitTask(OpTypes::kEqual, aops::EqualOpOptions{}, {*this, std::move(v)})[0];

Callers 1

Calls 1

deviceMethod · 0.45

Tested by

no test coverage detected