21 lines
508 B
Python
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()
|