Browse Source

[extension] fixed exception catch (#5342)

pull/5345/head
Frank Lee 10 months ago committed by GitHub
parent
commit
abd8e77ad8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      extensions/cpp_extension.py

2
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()

Loading…
Cancel
Save