Skip to content

Commit 4098dcf

Browse files
authored
Remove assert's relaxation in benchmarking code (#3826)
In the past, the error was for agama 1077: https://github.com/intel/intel-xpu-backend-for-triton/actions/runs/12915920053/job/36018883887. It's not reproducible for agama 1099 now: https://github.com/intel/intel-xpu-backend-for-triton/actions/runs/14242289577/job/39914776221?pr=3826. I guess I can remove the workaround and see how the code behaves in the future. Closes #3243 Signed-off-by: Anatoly Myachev <[email protected]>
1 parent e85b4f6 commit 4098dcf

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

benchmarks/triton_kernels_benchmark/benchmark_testing.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -159,8 +159,7 @@ def extract_kernels(funcs):
159159
# for correct registration of kernels.
160160
# For details: https://github.com/pytorch/pytorch/issues/144778
161161
kernels = [kernel for kernel in kernels if kernel != []]
162-
# FIXME: relaxation for new agama release
163-
assert len(kernels) >= n_repeat - 1, (
162+
assert len(kernels) == n_repeat, (
164163
f"the profiling number not match; {n_repeat=}, {kernels=}, \n" +
165164
f"top functions by xpu_time:\n {prof.key_averages(group_by_stack_n=5).table(sort_by='xpu_time')}")
166165
# Make the time to the milliseconds.

0 commit comments

Comments
 (0)