Merge branch 'main' into sync/main

2025-09-01 09:07:51 +00:00 · 2024-03-04 10:09:36 +08:00
parent 0aa27f1961 4b8312c08e
commit 0310b76e9d
80 changed files with 3464 additions and 1234 deletions
--- a/extensions/cpp_extension.py
+++ b/extensions/cpp_extension.py
@@ -126,7 +126,7 @@ class _CppExtension(_Extension):
    def load(self):
        try:
            op_kernel = self.import_op()
-        except ImportError:
+        except (ImportError, ModuleNotFoundError):
            # if import error occurs, it means that the kernel is not pre-built
            # so we build it jit
            op_kernel = self.build_jit()