Skip to content

Commit 96a5ba8

Browse files
[Experimental] Debug apparent race condition on top1
1 parent 7e56a94 commit 96a5ba8

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

python/benchmarks/python_overhead.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,9 @@ def matmul_bgrad(float(M,N) A, float(M,K) d_C) -> (d_B) {
111111
tuner_config)
112112
cache = MappingOptionsCache(cache_file.name)
113113
top10 = cache.load(mm, "matmul", (A, B), 10)
114-
assert top1.__str__() == top10[0].__str__()
114+
assert top1.__str__() == top10[0].__str__(), (
115+
"Expected {}\nGot {}\nTop10 are {}".format(
116+
top1, top10[0], "\n".join(opt for opt in top10)))
115117

116118
# Compile and run with the new options
117119
compilation_cache.compile("matmul", (A, B), top1)

0 commit comments

Comments
 (0)