ai-content-maker/.venv/Lib/site-packages/numba/cuda/tests/cudadrv/test_profiler.py

21 lines
508 B
Python

import unittest
from numba.cuda.testing import ContextResettingTestCase
from numba import cuda
from numba.cuda.testing import skip_on_cudasim
@skip_on_cudasim('CUDA Profiler unsupported in the simulator')
class TestProfiler(ContextResettingTestCase):
def test_profiling(self):
with cuda.profiling():
a = cuda.device_array(10)
del a
with cuda.profiling():
a = cuda.device_array(100)
del a
if __name__ == '__main__':
unittest.main()