from numba import cuda @cuda.jit(device=True) def cuda_module_in_device_function(): return cuda.threadIdx.x