示例#1
0
def torch_test(func):
    import taichi as ti
    if ti.has_pytorch():
        # OpenGL somehow crashes torch test without a reason, unforturnately
        return ti.archs_excluding(ti.opengl)(func)
    else:
        return lambda: None
示例#2
0
def archs_support_random(func):
    return ti.archs_excluding(ti.metal)(func)
示例#3
0
def ti_support_non_top_dynamic(test):
    return ti.archs_excluding(ti.opengl, ti.cc)(test)
示例#4
0
def ti_support_dynamic(test):
    return ti.archs_excluding(ti.cc)(test)
示例#5
0
def archs_support_bitmasked(func):
    return ti.archs_excluding(ti.opengl)(func)
示例#6
0
def all_archs_for_this(test):
    # ti.call_internal() is not supported on Metal and OpenGL yet
    return ti.archs_excluding(ti.metal, ti.opengl)(test)
示例#7
0
def all_archs_for_this(test):
    # ti.call_internal() is not supported on CUDA, Metal, OpenGL yet
    return ti.archs_excluding(ti.metal, ti.opengl, ti.cuda, ti.vulkan)(test)
示例#8
0
def ti_support_dynamic(test):
    return ti.archs_excluding(ti.opengl, ti.cc, ti.vulkan)(test)