性能分析器

Taichi 的分析器可以帮助你分析程序的运行时开销。在 Taichi 中有两个分析系统： ScopedProfiler 和 KernelProfiler.

基于作用域的性能分析器 - ScopedProfiler

  1. ScopedProfiler 能够有层次地度量 主机上 执行任务所花费的时间。
  2. 这个分析器是自动开启的。如要显示它的结果，请调用 ti.print_profile_info()。例如:
  1. import taichi as ti
  3. ti.init(arch=ti.cpu)
  4. var = ti.var(ti.f32, shape=1)
  7. @ti.kernel
  8. def compute():
  9.     var[0] = 1.0
  10.     print("Setting var[0] =", var[0])
  13. compute()
  14. ti.print_profile_info()

ti.print_profile_info() 会以有层次的格式打印分析结果。

注解

ScopedProfiler 是 Taichi 的一个核心 C++ 类。 它不会向 Python 用户公开。

内核性能分析器（KernelProfiler）

  1. KernelProfiler 记录了 Taichi 内核在设备上的开销。如要开启该分析器，请在 ti.init 中设置 kernel_profiler=True 。
  2. 通过调用 ti.kernel_profiler_print() 以显示内核分析器的结果。例如：
  1. import taichi as ti
  3. ti.init(ti.cpu, kernel_profiler=True)
  4. var = ti.var(ti.f32, shape=1)
  7. @ti.kernel
  8. def compute():
  9.     var[0] = 1.0
  12. compute()
  13. ti.kernel_profiler_print()

输出将会是：

  1. [ 22.73%] jit_evaluator_0_kernel_0_serial             min   0.001 ms   avg   0.001 ms   max   0.001 ms   total   0.000 s [      1x]
  2. [  0.00%] jit_evaluator_1_kernel_1_serial             min   0.000 ms   avg   0.000 ms   max   0.000 ms   total   0.000 s [      1x]
  3. [ 77.27%] compute_c4_0_kernel_2_serial                min   0.004 ms   avg   0.004 ms   max   0.004 ms   total   0.000 s [      1x]