예제 #1
0
def torch_test(func):
  import taichi as ti
  if ti.has_pytorch():
    return ti.all_archs(func)
  else:
    return lambda: None
예제 #2
0
def ti_support_dynamic(test):
    return ti.all_archs(test)