Use the SYCL event profiling time stamp to do the benchmark in softmax.

intel · May 16, 2024 · 81e6d06 · 81e6d06
1 parent 3acbc5d
commit 81e6d06
Show file tree

Hide file tree

Showing 7 changed files with 881 additions and 46 deletions.
diff --git a/benchmarks/xetla_benchmark/__init__.py b/benchmarks/xetla_benchmark/__init__.py
@@ -1,3 +1,12 @@
 import torch
 import intel_extension_for_pytorch
 from . import xetla_kernel
+from . import benchmark_testing
+from .benchmark_testing import do_bench, assert_close, perf_report, Benchmark
+
+import triton
+import triton.runtime.driver as driver
+from . import benchmark_driver
+
+# replace the launcher with the profilier hook.
+driver.active.launcher_cls = benchmark_driver.XPULauncher