question-mark
Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Cannot Import JIT optimized kernels?

See original GitHub issue

I 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:closed
  • Created a year ago
  • Comments:11 (5 by maintainers)

github_iconTop GitHub Comments

1reaction
Luodiancommented, Jul 13, 2022

I reinstall everything and it works fine. Thanks for your patience, it helps me a lot.

0reactions
ghostplantcommented, Jul 20, 2022

Thanks for your information. I’ll close this issue as it is solved.

Read more comments on GitHub >

github_iconTop 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 >

github_iconTop Related Medium Post

No results found

github_iconTop Related StackOverflow Question

No results found

github_iconTroubleshoot Live Code

Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free

github_iconTop Related Reddit Thread

No results found

github_iconTop Related Hackernoon Post

No results found

github_iconTop Related Tweet

No results found

github_iconTop Related Dev.to Post

No results found

github_iconTop Related Hashnode Post

No results found