Exemplo n.º 1
0
def torch_test(func):
  import taichi as ti
  if ti.has_pytorch():
    return ti.all_archs(func)
  else:
    return lambda: None
Exemplo n.º 2
0
def ti_support_dynamic(test):
    return ti.all_archs(test)