Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ColossalAI
Commits
abd8e77a
Unverified
Commit
abd8e77a
authored
Jan 31, 2024
by
Frank Lee
Committed by
GitHub
Jan 31, 2024
Browse files
[extension] fixed exception catch (#5342)
parent
71321a07
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
extensions/cpp_extension.py
extensions/cpp_extension.py
+1
-1
No files found.
extensions/cpp_extension.py
View file @
abd8e77a
...
...
@@ -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
()
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment