common.py 295 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12
"""Common utility for topi test"""

def get_all_backend():
    """return all supported target

    Returns
    -------
    targets: list
        A list of all supported targets
    """
    return ['llvm', 'cuda', 'opencl', 'metal', 'rocm', 'vulkan', 'nvptx',
            'llvm -device=arm_cpu']