Skip to content

Commit

Permalink
Another bugfix
Browse files Browse the repository at this point in the history
  • Loading branch information
xuzhao9 committed Nov 20, 2024
1 parent 88f0e68 commit cda4af0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tritonbench/utils/triton_op.py
Original file line number Diff line number Diff line change
Expand Up @@ -621,7 +621,7 @@ def __post__init__(self):

def _get_bm_func(self, bm_func_name: str):
fwd_fn_lambda = getattr(self, bm_func_name, None)
assert fwd_fn_lambda, (
assert callable(fwd_fn_lambda), (
f"Could not find benchmark {bm_func_name} registered in {self.name}. "
f"Available benchmarks: {REGISTERED_BENCHMARKS[self.name].keys()}. "
)
Expand Down

0 comments on commit cda4af0

Please sign in to comment.