fix external test + speed

This commit is contained in:
George Hotz
2023-03-03 06:46:16 -08:00
parent 8919ca8163
commit 999b44c274
2 changed files with 10 additions and 2 deletions

View File

@@ -132,7 +132,7 @@ jobs:
- name: Install Dependencies
run: pip install -e '.[gpu,testing]' --extra-index-url https://download.pytorch.org/whl/cpu
- name: Run Optimizer Test
run: GPU=1 python test/external_test_opt.py
run: GPU=1 python test/external/external_test_opt.py
- name: Run Pytest (default)
run: GPU=1 python -m pytest -s -v -n=auto