Function check_numba_compatibility_gpu

dali/test/python/test_utils.py:962–976

(if_skip=True, use_experimental: bool = False)

def check_numba_compatibility_gpu(if_skip=True, use_experimental: bool = False):
    if use_experimental:
        from nvidia.dali.plugin.numba.experimental import NumbaFunction
    else:
        from nvidia.dali.plugin.numba import NumbaFunction

    if not NumbaFunction._check_minimal_numba_version(
        False
    ) or not NumbaFunction._check_cuda_compatibility(False):
        if if_skip:
            raise SkipTest()
        else:
            return False
    if not if_skip:
        return True


def create_sign_off_registry():