Skip to content

Commit 4e41541

Browse files
authored
1 parent 4355afd commit 4e41541

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

.github/pins/pytorch-upstream.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
0a2685160140656e3e53818611dd2c65c4397be5
1+
8321eec009c8c79145ebccd51fdfc336e5f8b848

python/triton/runtime/autotuner.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,10 @@ def _post_hook(args, exception):
9090
while not inspect.isfunction(self.base_fn):
9191
self.base_fn = self.base_fn.fn
9292

93+
self.num_warmups = warmup
94+
self.num_reps = rep
95+
self.use_cuda_graph = use_cuda_graph
96+
9397
# If we got explicitly called via the old interface, raise a warning
9498
# and proceed with the old behavior.
9599
if warmup is not None or rep is not None or use_cuda_graph:

0 commit comments

Comments
 (0)