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

55 lines
1.5 KiB
Python
Raw Normal View History

2024-05-03 04:18:51 +03:00
import multiprocessing as mp
import traceback
from numba.cuda.testing import unittest, CUDATestCase
from numba.cuda.testing import (skip_on_cudasim, skip_under_cuda_memcheck,
skip_if_mvc_libraries_unavailable)
from numba.tests.support import linux_only
def child_test():
from numba import config, cuda
# Change the MVC config after importing numba.cuda
config.CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY = 1
@cuda.jit
def f():
pass
f[1, 1]()
def child_test_wrapper(result_queue):
try:
output = child_test()
success = True
# Catch anything raised so it can be propagated
except: # noqa: E722
output = traceback.format_exc()
success = False
result_queue.put((success, output))
@linux_only
@skip_under_cuda_memcheck('May hang CUDA memcheck')
@skip_on_cudasim('Simulator does not require or implement MVC')
@skip_if_mvc_libraries_unavailable
class TestMinorVersionCompatibility(CUDATestCase):
def test_mvc(self):
# Run test with Minor Version Compatibility enabled in a child process
ctx = mp.get_context('spawn')
result_queue = ctx.Queue()
proc = ctx.Process(target=child_test_wrapper, args=(result_queue,))
proc.start()
proc.join()
success, output = result_queue.get()
# Ensure the child process ran to completion before checking its output
if not success:
self.fail(output)
if __name__ == '__main__':
unittest.main()