()
| 34 | |
| 35 | |
| 36 | def _get_kernel(): |
| 37 | global _kernel, _kernel_load_failed |
| 38 | if _kernel_load_failed: |
| 39 | return None |
| 40 | if _kernel is not None: |
| 41 | return _kernel |
| 42 | try: |
| 43 | from kernels import get_kernel |
| 44 | |
| 45 | _kernel = get_kernel("kernels-community/bitsandbytes-mps", version=1) |
| 46 | except Exception: |
| 47 | _kernel_load_failed = True |
| 48 | return None |
| 49 | return _kernel |
| 50 | |
| 51 | |
| 52 | @_try_torch_compile(dynamic=True) |
no outgoing calls
no test coverage detected