Cannot Import JIT optimized kernels?
See original GitHub issueI used TUTEL for a while and it works greatly fine. But today I updated my environment and reinstall tutel, I found it crashed during importing module. Do you have any idea on why this happen? Thanks!
>>> from tutel import moe as tutel_moe
Traceback (most recent call last):
File "/mnt/lustre/bli/anaconda3/envs/scale/lib/python3.9/site-packages/tutel/impls/jit_compiler.py", line 8, in <module>
import tutel_custom_kernel
ImportError: /mnt/lustre/bli/anaconda3/envs/scale/lib/python3.9/site-packages/tutel_custom_kernel.cpython-39-x86_64-linux-gnu.so: undefined symbol: _ZNSt15__exception_ptr13exception_ptr9_M_addrefEv
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/mnt/lustre/bli/anaconda3/envs/scale/lib/python3.9/site-packages/tutel/moe.py", line 6, in <module>
from .jit_kernels.gating import fast_cumsum_sub_one
File "/mnt/lustre/bli/anaconda3/envs/scale/lib/python3.9/site-packages/tutel/jit_kernels/gating.py", line 7, in <module>
from ..impls.jit_compiler import tutel_custom_kernel
File "/mnt/lustre/bli/anaconda3/envs/scale/lib/python3.9/site-packages/tutel/impls/jit_compiler.py", line 10, in <module>
raise Exception("Cannot import JIT optimized kernels. Did you forget to install Custom Kernel Extension?")
Exception: Cannot import JIT optimized kernels. Did you forget to install Custom Kernel Extension?
Issue Analytics
- State:
- Created a year ago
- Comments:11 (5 by maintainers)
Top Results From Across the Web
Compiling Python code with @jit - Numba documentation
Whenever Numba optimizes Python code to native code that only works on native types and variables (rather than Python objects), it is not...
Read more >Installing a Custom Kernel Extension - Apple Developer
The most common way to install kexts is with a custom installer package. (Unlike apps and system extensions, you cannot distribute kexts through...
Read more >2- Custom CUDA Kernels in Python with Numba | Kaggle
CUDA kernels are compiled using the numba.cuda.jit decorator. numba.cuda.jit is not to be ... from numba import cuda # Note the use of...
Read more >Running Python UDFs in Native NVIDIA CUDA Kernels with ...
This framework uses the Numba Python compiler and Jitify CUDA just-in-time (JIT) compilation library to provide cuDF users the flexibility ...
Read more >Writing custom CUDA kernels with Triton - Kushajveer Singh
PyTorch JIT is one option in these cases. But PyTorch JIT is a high-level compiler that can only optimize parts of the code...
Read more >
Top Related Medium Post
No results found
Top Related StackOverflow Question
No results found
Troubleshoot Live Code
Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free
Top Related Reddit Thread
No results found
Top Related Hackernoon Post
No results found
Top Related Tweet
No results found
Top Related Dev.to Post
No results found
Top Related Hashnode Post
No results found
I reinstall everything and it works fine. Thanks for your patience, it helps me a lot.
Thanks for your information. I’ll close this issue as it is solved.