Function check_numba_compatibility_cpu

dali/test/python/test_utils.py:942–959

(if_skip=True)

def check_numba_compatibility_cpu(if_skip=True):
    import numba

    # There's a bug in LLVM JIT linker that makes the tests fail
    # randomly on 64-bit ARM platform for some NUMBA versions.
    #
    # Numba bug:
    # https://github.com/numba/numba/issues/8567
    if platform.processor().lower() in ("arm64", "aarch64", "armv8") and (
        Version(numba.__version__) >= Version("0.57.0")
        and Version(numba.__version__) < Version("0.59.0")
    ):
        if if_skip:
            raise SkipTest()
        else:
            return False
    if not if_skip:
        return True


def check_numba_compatibility_gpu(if_skip=True, use_experimental: bool = False):