Fix CI failures

xuzhao9 · web-flow · commit cbca9223871c · 2025-11-03T17:53:45.000Z
Differential Revision: D86037092 Pull Request resolved: #613
diff --git a/tritonbench/utils/triton_op.py b/tritonbench/utils/triton_op.py
@@ -891,6 +891,9 @@ def __post__init__(self):
             f"Input IDs to run: {self._input_ids}",
         )
 
+    def get_backend(self, bm_func_name: str):
+        return REGISTERED_BENCHMARKS[self.name][bm_func_name]
+
     def _get_bm_func(self, bm_func_name: str):
         fwd_fn_lambda = getattr(self, bm_func_name, None)
         assert callable(fwd_fn_lambda), (

Original file line number	Diff line number	Diff line change
`@@ -891,6 +891,9 @@ def __post__init__(self):`
`891`	`891`	`f"Input IDs to run: {self._input_ids}",`
`892`	`892`	`)`
`893`	`893`
	`894`	`+ def get_backend(self, bm_func_name: str):`
	`895`	`+ return REGISTERED_BENCHMARKS[self.name][bm_func_name]`
	`896`	`+`
`894`	`897`	`def _get_bm_func(self, bm_func_name: str):`
`895`	`898`	`fwd_fn_lambda = getattr(self, bm_func_name, None)`
`896`	`899`	`assert callable(fwd_fn_lambda), (`