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()
def test(*args, **kwargs): archs = [_ti_core.host_arch()] for arch in archs: ti.init(arch=arch) func(*args, **kwargs)
def get_host_arch_list(): return [_ti_core.host_arch()]