sglang_v0.5.2/pytorch_2.8.0/torchgen/_autoheuristic/test.sh

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 ../