benchmark
66e1f3d3 - treat like a template in GEMMs (#161342)

Commit
103 days ago
treat like a template in GEMMs (#161342) Summary: # why - central point to analyze and override all generated choices # what - add a pseudo heuristic for aten that just yields a single, empty kwargs - add a pseudo heuristic with the bias_addmm logic for it - add an addmm specific heuristic that yields a single choice, but also expands it with alpha and beta kwargs - replace all the aten.bind calls with V.choices.get_mm_configs using the now matching API for aten # testing ``` python3 -bb -m pytest test/inductor/test_max_autotune.py -v ``` X-link: https://github.com/pytorch/pytorch/pull/161342 Approved by: https://github.com/jansel ghstack dependencies: #162075, #161340, #161341 Reviewed By: jeanschmidt Differential Revision: D81520580 fbshipit-source-id: cd98fbfc056e8d8c7145ef5963e31bafbde52a0e
Author
Parents
Loading