pytorch
694ed70e - [inductor][easy] create a wrap for triton do_bench function (#99216)

Commit
1 year ago
[inductor][easy] create a wrap for triton do_bench function (#99216) triton PR https://github.com/openai/triton/pull/1513 change the interface of do_bench function. The quantile fields name is changed from 'percentiles' to 'quantiles' and its default value is changed from from (0.5, 0.2, 0.8) to None. This break some inductor code since a caller expects a tuple may get a item. Add a wrapper to maintain the same behavior for inductor. Pull Request resolved: https://github.com/pytorch/pytorch/pull/99216 Approved by: https://github.com/jansel, https://github.com/ngimel
Author
Committer
Parents
Loading