Exemple #1
0
cross = deprecated('ti.cross(a, b)', 'a.cross(b)')(Matrix.cross)
dot = deprecated('ti.dot(a, b)', 'a.dot(b)')(Matrix.dot)
normalized = deprecated('ti.normalized(a)',
                        'a.normalized()')(Matrix.normalized)

cfg = default_cfg()
x86_64 = _ti_core.x64
x64 = _ti_core.x64
arm64 = _ti_core.arm64
cuda = _ti_core.cuda
metal = _ti_core.metal
opengl = _ti_core.opengl
cc = _ti_core.cc
wasm = _ti_core.wasm
gpu = [cuda, metal, opengl]
cpu = _ti_core.host_arch()
kernel_profiler_print = lambda: impl.get_runtime().prog.kernel_profiler_print()
kernel_profiler_clear = lambda: impl.get_runtime().prog.kernel_profiler_clear()
kernel_profiler_total_time = lambda: impl.get_runtime(
).prog.kernel_profiler_total_time()
timeline_clear = lambda: impl.get_runtime().prog.timeline_clear()
timeline_save = lambda fn: impl.get_runtime().prog.timeline_save(fn)

# Legacy API
type_factory_ = _ti_core.get_type_factory_instance()


def memory_profiler_print():
    impl.get_runtime().materialize()
    impl.get_runtime().prog.print_memory_profiler_info()
Exemple #2
0
 def test(*args, **kwargs):
     archs = [_ti_core.host_arch()]
     for arch in archs:
         ti.init(arch=arch)
         func(*args, **kwargs)
Exemple #3
0
def get_host_arch_list():
    return [_ti_core.host_arch()]