InfiniTensor · voltjia · Oct 13, 2025 · Oct 13, 2025 · Oct 13, 2025
diff --git a/src/ntops/__init__.py b/src/ntops/__init__.py
@@ -1,3 +1,3 @@
-import torch
+from ntops import torch
 
 __all__ = ["torch"]
diff --git a/tests/test_abs.py b/tests/test_abs.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_add.py b/tests/test_add.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import gauss, generate_arguments
 

diff --git a/tests/test_addmm.py b/tests/test_addmm.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.test_mm import generate_arguments
 from tests.utils import gauss

diff --git a/tests/test_bitwise_and.py b/tests/test_bitwise_and.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_bitwise_not.py b/tests/test_bitwise_not.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_bitwise_or.py b/tests/test_bitwise_or.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_bmm.py b/tests/test_bmm.py
@@ -3,7 +3,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.test_mm import generate_arguments
 

diff --git a/tests/test_clamp.py b/tests/test_clamp.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_cos.py b/tests/test_cos.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_div.py b/tests/test_div.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_dropout.py b/tests/test_dropout.py
@@ -4,7 +4,7 @@
 import torch
 import torch.nn.functional as F
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_eq.py b/tests/test_eq.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_exp.py b/tests/test_exp.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_ge.py b/tests/test_ge.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_gelu.py b/tests/test_gelu.py
@@ -2,7 +2,7 @@
 import torch
 import torch.nn.functional as F
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_gt.py b/tests/test_gt.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_isinf.py b/tests/test_isinf.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_isnan.py b/tests/test_isnan.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_layer_norm.py b/tests/test_layer_norm.py
@@ -3,7 +3,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_le.py b/tests/test_le.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_lt.py b/tests/test_lt.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_mm.py b/tests/test_mm.py
@@ -3,7 +3,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 
 

diff --git a/tests/test_mul.py b/tests/test_mul.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_ne.py b/tests/test_ne.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_neg.py b/tests/test_neg.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_pow.py b/tests/test_pow.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_relu.py b/tests/test_relu.py
@@ -2,7 +2,7 @@
 import torch
 import torch.nn.functional as F
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_rms_norm.py b/tests/test_rms_norm.py
@@ -3,7 +3,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_rotary_position_embedding.py b/tests/test_rotary_position_embedding.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 
 

diff --git a/tests/test_rsqrt.py b/tests/test_rsqrt.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_scaled_dot_product_attention.py b/tests/test_scaled_dot_product_attention.py
@@ -7,7 +7,7 @@
 import torch.nn.functional as F
 from torch.nn.attention.bias import causal_lower_right
 
-import ntops.torch
+import ntops
 from ntops.kernels.scaled_dot_product_attention import CausalVariant
 from tests.skippers import skip_if_cuda_not_available
 

diff --git a/tests/test_sigmoid.py b/tests/test_sigmoid.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_silu.py b/tests/test_silu.py
@@ -2,7 +2,7 @@
 import torch
 import torch.nn.functional as F
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_sin.py b/tests/test_sin.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_softmax.py b/tests/test_softmax.py
@@ -3,7 +3,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments
 

diff --git a/tests/test_sub.py b/tests/test_sub.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import gauss, generate_arguments
 

diff --git a/tests/test_tanh.py b/tests/test_tanh.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-import ntops.torch
+import ntops
 from tests.skippers import skip_if_cuda_not_available
 from tests.utils import generate_arguments