diff --git a/.github/pins/pytorch-upstream.txt b/.github/pins/pytorch-upstream.txt
index 67bed0a666..4cf70cf66d 100644
--- a/.github/pins/pytorch-upstream.txt
+++ b/.github/pins/pytorch-upstream.txt
@@ -1 +1 @@
-0a2685160140656e3e53818611dd2c65c4397be5
+8321eec009c8c79145ebccd51fdfc336e5f8b848
diff --git a/python/triton/runtime/autotuner.py b/python/triton/runtime/autotuner.py
index 5f846de170..be02d61a43 100644
--- a/python/triton/runtime/autotuner.py
+++ b/python/triton/runtime/autotuner.py
@@ -90,6 +90,10 @@ def _post_hook(args, exception):
         while not inspect.isfunction(self.base_fn):
             self.base_fn = self.base_fn.fn
 
+        self.num_warmups = warmup
+        self.num_reps = rep
+        self.use_cuda_graph = use_cuda_graph
+
         # If we got explicitly called via the old interface, raise a warning
         # and proceed with the old behavior.
         if warmup is not None or rep is not None or use_cuda_graph: