Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Report tflops by default for gemm; fix exception handling (pytorch#2259)
Summary: TFLOPS is the core metric for gemm. Along the way I hit some bugs and weirdness: - You couldn't Ctrl-C out of tritonbench, because the `finally` clause contained a return, which [suppresses the exception](https://docs.python.org/3/tutorial/errors.html#defining-clean-up-actions) - In generally I don't think the framework should catch RuntimeErrors, it makes it really hard to debug stuff because the desired result just ends up missing - In fact we had a typo (`metric` instead of `metrics` in the framework code that was never caught because it was caught and suppressed Pull Request resolved: pytorch#2259 Test Plan: ``` python run_benchmark.py triton --op gemm --splitk ``` Reviewed By: xuzhao9 Differential Revision: D57171806 Pulled By: bertmaher fbshipit-source-id: 74568625ad10907d9def8916abfc2f6292cdc6d6
- Loading branch information