ai-content-maker/.venv/Lib/site-packages/numba/cuda/tests/cudasim/support.py

7 lines
114 B
Python
Raw Normal View History

2024-05-03 04:18:51 +03:00
from numba import cuda
@cuda.jit(device=True)
def cuda_module_in_device_function():
return cuda.threadIdx.x