def torch_test(func): import taichi as ti if ti.has_pytorch(): return ti.all_archs(func) else: return lambda: None
def ti_support_dynamic(test): return ti.all_archs(test)