Skip to content

Conversation

@yeahdongcn
Copy link
Collaborator

Testing Done

All 244 tests pass on both test containers (torch_musa 2.7.0 and 2.7.1).

root@xiaodongye-s80:/ws# TORCHADA_ENABLE_CPP_OPS=1 TORCHADA_CPP_OPS_VERBOSE=1 python -c "import torchada"
Using /root/.cache/torch_extensions/py310_mu40304 as PyTorch extensions root...
Detected MUSA files, patching ldflags
Emitting ninja build file /root/.cache/torch_extensions/py310_mu40304/torchada_cpp_ops/build.ninja...
Building extension module torchada_cpp_ops...
Using envvar MAX_JOBS (12) as the number of workers...
ninja: no work to do.
Loading extension module torchada_cpp_ops...
root@xiaodongye-s80:/ws# TORCHADA_ENABLE_CPP_OPS=1 TORCHADA_DEBUG_CPP_OPS=1 python -c "
import torch
import torchada

x = torch.tensor([1.0, -2.0, 3.0], device='cuda')
y = torch.neg(x)
print('Result:', y.cpu())
"
[torchada] neg called
Result: tensor([-1.,  2., -3.])

Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com>
@yeahdongcn
Copy link
Collaborator Author

@wenqf11 PTAL.

Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com>
Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants