Skip to content

Commit f53a0f9

Browse files
desertfirepytorchmergebot
authored andcommitted
[Inductor] Fix test_profiler_mark_wrapper_call_cuda_cuda_wrapper (pytorch#136356)
Summary: Internal profiler behaves differently after turning on triton.autotune_at_compile_time. Needs more investigation but turning it off for this test for now. Reviewed By: henrylhtsang Differential Revision: D63035855 Pull Request resolved: pytorch#136356 Approved by: https://github.com/henrylhtsang
1 parent 5997354 commit f53a0f9

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/inductor/test_torchinductor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9514,6 +9514,7 @@ def fn(n, a):
95149514
self.assertEqual(x[0], -1)
95159515
self.assertEqual(cnts.frame_count, frame_count + 1)
95169516

9517+
@config.patch({"triton.autotune_at_compile_time": False})
95179518
@config.patch(profiler_mark_wrapper_call=True)
95189519
def test_profiler_mark_wrapper_call(self):
95199520
from torch.profiler import profile

0 commit comments

Comments
 (0)