issue/561: 将调用 ntops 算子设为默认

* 以变量形式使用 `use_ntops`

* 移除不需要的 `ntops.py`

Author: voltjia

python/infinicore/__init__.py

@@ -1,3 +1,5 @@
+import contextlib
+
 from infinicore.device import device
 from infinicore.dtype import (
     bfloat16,
@@ -24,7 +26,6 @@
     short,
     uint8,
 )
-from infinicore.ntops import use_ntops
 from infinicore.ops.add import add
 from infinicore.ops.attention import attention
 from infinicore.ops.matmul import matmul
@@ -68,8 +69,6 @@
     "long",
     "short",
     "uint8",
-    # `ntops` integration.
-    "use_ntops",
     # Operations.
     "add",
     "attention",
@@ -82,3 +81,15 @@
     "strided_from_blob",
     "zeros",
 ]
+
+use_ntops = False
+
+with contextlib.suppress(ImportError, ModuleNotFoundError):
+    import sys
+
+    import ntops
+
+    for op_name in ntops.torch.__all__:
+        getattr(ntops.torch, op_name).__globals__["torch"] = sys.modules[__name__]
+
+    use_ntops = True