Skip to content
Go to file
Cannot retrieve contributors at this time
12 lines (9 sloc) 377 Bytes
def init_jit():
from numba.cuda.compiler import Dispatcher
return Dispatcher
def initialize_all():
# Import models to register them with the data model manager
import numba.cuda.models # noqa: F401
from numba.core.registry import dispatcher_registry
dispatcher_registry.ondemand['gpu'] = init_jit
dispatcher_registry.ondemand['cuda'] = init_jit
You can’t perform that action at this time.