fix handcode_opt script (#5435)

* fix handcode_opt script

* run in ci

* real run in ci

* HALF=0
This commit is contained in:
chenyu
2024-07-12 20:52:28 -04:00
committed by GitHub
parent b80fd7d23c
commit 9a187e6102
2 changed files with 9 additions and 5 deletions

View File

@@ -247,6 +247,9 @@ jobs:
- if: ${{ matrix.task == 'onnx' }}
name: Test THREEFRY
run: PYTHONPATH=. THREEFRY=1 GPU=1 python3 -m pytest test/test_randomness.py test/test_jit.py --durations=20
- if: ${{ matrix.task == 'onnx' }}
name: Run handcode_opt
run: PYTHONPATH=. MODEL=resnet GPU=1 DEBUG=1 BS=4 HALF=0 python3 examples/handcode_opt.py
- name: Run process replay tests
if: env.RUN_PROCESS_REPLAY == '1'
run: cp test/external/process_replay/process_replay.py ./process_replay.py && git fetch origin master && git -c advice.detachedHead=false checkout origin/master && PYTHONPATH=. python3 process_replay.py