14 lines
546 B
Python
14 lines
546 B
Python
def initialize_all():
|
|
# Import models to register them with the data model manager
|
|
import numba.cuda.models # noqa: F401
|
|
|
|
from numba.cuda.decorators import jit
|
|
from numba.cuda.dispatcher import CUDADispatcher
|
|
from numba.core.target_extension import (target_registry,
|
|
dispatcher_registry,
|
|
jit_registry)
|
|
|
|
cuda_target = target_registry["cuda"]
|
|
jit_registry[cuda_target] = jit
|
|
dispatcher_registry[cuda_target] = CUDADispatcher
|