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()