Skip to content

Commit 4f60f45

Browse files
authored
Print exception message when accuracy check fails (#492)
* Print exception message when accuracy check fails * use logger.warning
1 parent 38e639a commit 4f60f45

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tritonbench/utils/triton_op.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1480,8 +1480,8 @@ def accuracy(self, fn: Callable, baseline_fn: Callable) -> bool:
14801480
atol=self.tb_args.atol,
14811481
)
14821482
return True
1483-
except Exception:
1484-
# either the output tensor or the grad tensor does not match
1483+
except Exception as e:
1484+
logger.warning(f"Exception during accuracy check: {e}")
14851485
return False
14861486

14871487
def _do_bench(

0 commit comments

Comments
 (0)