mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
5dfb22d4c8
This PR adds tests to AutoHeuristic that ensure that when existing heuristics are re-generated, the generated code stays the same. Pull Request resolved: https://github.com/pytorch/pytorch/pull/133496 Approved by: https://github.com/eellison
8 lines
245 B
Bash
8 lines
245 B
Bash
# you should run these tests whenever you make changes to any of the train*.py files within this directory
|
|
# running these tests takes around 10 minutes on my machine
|
|
cd mixed_mm
|
|
python test_mixed_mm.py
|
|
cd ../pad_mm
|
|
python test_pad_mm.py
|
|
cd ../
|