Update __init__.py,'th.device("cuda") ' 会导致test_edge_softmax.py 等测例出现较多的device不匹配的报错，暂时还原

Update init.py,'th.device("cuda") ' 会导致test_edge_softmax.py 等测例出现较多的device不匹配的报错，暂时还原
187b354a · sangwz · 46fc0f71 · 187b354a
Commit 187b354a authored Sep 01, 2025 by sangwz
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/backend/pytorch/__init__.py tests/backend/pytorch/__init__.py +1 -1

No files found.
--- a/tests/backend/pytorch/__init__.py
+++ b/tests/backend/pytorch/__init__.py
@@ -4,7 +4,7 @@ import torch as th


 def cuda():
-    return th.device("cuda")
+    return th.device("cuda:0")


 def is_cuda_available():