Exit scope

pytorch-labs · Dec 7, 2024 · edc923f · edc923f
1 parent 38184b2
commit edc923f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tritonbench/utils/triton_op.py b/tritonbench/utils/triton_op.py
@@ -786,11 +786,11 @@ def _reduce_benchmarks(acc, bm_name: str):
                 del self.example_inputs  # save some memory
                 if "proton" in self.required_metrics:
                     proton.activate(self._proton_session_id)
-                    proton.exit_scope(f"x_val_{x_val}")
+                    proton.exit_scope()
                     proton.deactivate(self._proton_session_id)
             if "proton" in self.required_metrics:
                 proton.activate(self._proton_session_id)
-                proton.exit_scope(f"tritonbench_run_op_{self.name}")
+                proton.exit_scope()
                 proton.finalize()
         except (KeyboardInterrupt, Exception):
             logger.warning(