Switch from import torch to from ntops import torch

[voltjia]

pytest output:

============================= test session starts ==============================
platform linux -- Python 3.10.12, pytest-8.3.3, pluggy-1.5.0
rootdir: /home/huangjiacheng/ntops
configfile: pyproject.toml
plugins: cov-6.0.0
collected 636 items

tests/test_abs.py ........                                               [  1%]
tests/test_add.py ........                                               [  2%]
tests/test_addmm.py ..                                                   [  2%]
tests/test_bitwise_and.py ................                               [  5%]
tests/test_bitwise_not.py ................                               [  7%]
tests/test_bitwise_or.py ................                                [ 10%]
tests/test_bmm.py ..                                                     [ 10%]
tests/test_clamp.py ........                                             [ 11%]
tests/test_cos.py ........                                               [ 13%]
tests/test_div.py ........                                               [ 14%]
tests/test_dropout.py ........                                           [ 15%]
tests/test_eq.py ........                                                [ 16%]
tests/test_exp.py ........                                               [ 18%]
tests/test_ge.py ........                                                [ 19%]
tests/test_gelu.py ........                                              [ 20%]
tests/test_gt.py ........                                                [ 22%]
tests/test_isinf.py ........                                             [ 23%]
tests/test_isnan.py ........                                             [ 24%]
tests/test_layer_norm.py ............................................... [ 31%]
.................................................                        [ 39%]
tests/test_le.py ........                                                [ 40%]
tests/test_lt.py ........                                                [ 42%]
tests/test_mm.py ..                                                      [ 42%]
tests/test_mul.py ........                                               [ 43%]
tests/test_ne.py ........                                                [ 44%]
tests/test_neg.py ........                                               [ 46%]
tests/test_pow.py ........                                               [ 47%]
tests/test_relu.py ........                                              [ 48%]
tests/test_rms_norm.py ................................................. [ 56%]
...............                                                          [ 58%]
tests/test_rotary_position_embedding.py ................................ [ 63%]
........................................................................ [ 75%]
........................                                                 [ 78%]
tests/test_rsqrt.py ........                                             [ 80%]
tests/test_scaled_dot_product_attention.py ............................. [ 84%]
.................................................                        [ 92%]
tests/test_sigmoid.py ........                                           [ 93%]
tests/test_silu.py ........                                              [ 94%]
tests/test_sin.py ........                                               [ 96%]
tests/test_softmax.py ........                                           [ 97%]
tests/test_sub.py ........                                               [ 98%]
tests/test_tanh.py ........                                              [100%]

=============================== warnings summary ===============================
tests/test_isinf.py: 24 warnings
  /home/huangjiacheng/.ninetoothed/764184e22db5a83a8e6c05ac1bf3cf51e19c0aef4811498b47bd12af26f61972.py:21: UserWarning: Logical operators 'and' and 'or' are deprecated for non-scalar tensors; please use '&' or '|' instead
    triton.language.store(ninetoothed_tensor_741_pointers + (ninetoothed_tensor_741_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_69 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_69)[::,]) * ninetoothed_tensor_741_stride_0, pos_result or neg_result, mask=(ninetoothed_tensor_741_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_69 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_69)[::,] < ninetoothed_tensor_741_size_0) & (triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_69)[::,] < ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_69))
  Enable tracemalloc to get traceback where the object was allocated.
  See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.

tests/test_isinf.py: 24 warnings
  /home/huangjiacheng/.ninetoothed/dd9bf311398ea0add547c72852fb2ecb466e235f1933ce4e70146b1baf90618b.py:21: UserWarning: Logical operators 'and' and 'or' are deprecated for non-scalar tensors; please use '&' or '|' instead
    triton.language.store(ninetoothed_tensor_749_pointers + ((ninetoothed_tensor_749_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_70 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_70)[::,]) // ninetoothed_tensor_749_size_1 * ninetoothed_tensor_749_stride_0 + (ninetoothed_tensor_749_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_70 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_70)[::,]) % ninetoothed_tensor_749_size_1 * ninetoothed_tensor_749_stride_1), pos_result or neg_result, mask=((ninetoothed_tensor_749_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_70 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_70)[::,]) // ninetoothed_tensor_749_size_1 < ninetoothed_tensor_749_size_0) & ((ninetoothed_tensor_749_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_70 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_70)[::,]) % ninetoothed_tensor_749_size_1 < ninetoothed_tensor_749_size_1) & (triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_70)[::,] < ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_70))
  Enable tracemalloc to get traceback where the object was allocated.
  See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.

tests/test_isinf.py: 24 warnings
  /home/huangjiacheng/.ninetoothed/e85c1c3c1a2674d004bef3e21aea3488fc27e8829df2ce4e83f4b720f843ffee.py:21: UserWarning: Logical operators 'and' and 'or' are deprecated for non-scalar tensors; please use '&' or '|' instead
    triton.language.store(ninetoothed_tensor_757_pointers + ((ninetoothed_tensor_757_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71)[::,]) // (ninetoothed_tensor_757_size_1 * ninetoothed_tensor_757_size_2) * ninetoothed_tensor_757_stride_0 + (ninetoothed_tensor_757_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71)[::,]) % (ninetoothed_tensor_757_size_1 * ninetoothed_tensor_757_size_2) // ninetoothed_tensor_757_size_2 * ninetoothed_tensor_757_stride_1 + (ninetoothed_tensor_757_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71)[::,]) % (ninetoothed_tensor_757_size_1 * ninetoothed_tensor_757_size_2) % ninetoothed_tensor_757_size_2 * ninetoothed_tensor_757_stride_2), pos_result or neg_result, mask=((ninetoothed_tensor_757_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71)[::,]) // (ninetoothed_tensor_757_size_1 * ninetoothed_tensor_757_size_2) < ninetoothed_tensor_757_size_0) & ((ninetoothed_tensor_757_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71)[::,]) % (ninetoothed_tensor_757_size_1 * ninetoothed_tensor_757_size_2) // ninetoothed_tensor_757_size_2 < ninetoothed_tensor_757_size_1) & ((ninetoothed_tensor_757_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71)[::,]) % (ninetoothed_tensor_757_size_1 * ninetoothed_tensor_757_size_2) % ninetoothed_tensor_757_size_2 < ninetoothed_tensor_757_size_2) & (triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71)[::,] < ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_71))
  Enable tracemalloc to get traceback where the object was allocated.
  See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.

tests/test_isinf.py: 24 warnings
  /home/huangjiacheng/.ninetoothed/39a5dd979812f81f1391a30e2934e08b741184a2ed913ab4ddcfa574acbeace3.py:21: UserWarning: Logical operators 'and' and 'or' are deprecated for non-scalar tensors; please use '&' or '|' instead
    triton.language.store(ninetoothed_tensor_765_pointers + ((ninetoothed_tensor_765_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72)[::,]) // (ninetoothed_tensor_765_size_1 * (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3)) * ninetoothed_tensor_765_stride_0 + (ninetoothed_tensor_765_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72)[::,]) % (ninetoothed_tensor_765_size_1 * (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3)) // (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3) * ninetoothed_tensor_765_stride_1 + (ninetoothed_tensor_765_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72)[::,]) % (ninetoothed_tensor_765_size_1 * (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3)) % (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3) // ninetoothed_tensor_765_size_3 * ninetoothed_tensor_765_stride_2 + (ninetoothed_tensor_765_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72)[::,]) % (ninetoothed_tensor_765_size_1 * (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3)) % (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3) % ninetoothed_tensor_765_size_3 * ninetoothed_tensor_765_stride_3), pos_result or neg_result, mask=((ninetoothed_tensor_765_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72)[::,]) // (ninetoothed_tensor_765_size_1 * (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3)) < ninetoothed_tensor_765_size_0) & ((ninetoothed_tensor_765_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72)[::,]) % (ninetoothed_tensor_765_size_1 * (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3)) // (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3) < ninetoothed_tensor_765_size_1) & ((ninetoothed_tensor_765_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72)[::,]) % (ninetoothed_tensor_765_size_1 * (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3)) % (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3) // ninetoothed_tensor_765_size_3 < ninetoothed_tensor_765_size_2) & ((ninetoothed_tensor_765_index_0[::,] * ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72 + triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72)[::,]) % (ninetoothed_tensor_765_size_1 * (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3)) % (ninetoothed_tensor_765_size_2 * ninetoothed_tensor_765_size_3) % ninetoothed_tensor_765_size_3 < ninetoothed_tensor_765_size_3) & (triton.language.arange(0, ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72)[::,] < ninetoothed_meta_prefix_ninetoothed_BLOCK_SIZE_72))
  Enable tracemalloc to get traceback where the object was allocated.
  See https://docs.pytest.org/en/stable/how-to/capture-warnings.html#resource-warnings for more info.

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
================ 636 passed, 96 warnings in 3530.44s (0:58:50) =================