Skip to content

Commit

Permalink
install requirements
Browse files Browse the repository at this point in the history
  • Loading branch information
zewenli98 committed Jan 16, 2025
1 parent afa9547 commit 371fe76
Showing 1 changed file with 18 additions and 6 deletions.
24 changes: 18 additions & 6 deletions .github/workflows/build-test-linux.yml
Original file line number Diff line number Diff line change
Expand Up @@ -136,7 +136,9 @@ jobs:
export USE_HOST_DEPS=1
export CI_BUILD=1
pushd .
cd tests/py/dynamo
cd tests/py
python -m pip install -r requirements.txt
cd dynamo
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/
popd
Expand Down Expand Up @@ -165,7 +167,9 @@ jobs:
export USE_HOST_DEPS=1
export CI_BUILD=1
pushd .
cd tests/py/dynamo
cd tests/py
python -m pip install -r requirements.txt
cd dynamo
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/
popd
Expand Down Expand Up @@ -194,7 +198,9 @@ jobs:
export USE_HOST_DEPS=1
export CI_BUILD=1
pushd .
cd tests/py/dynamo
cd tests/py
python -m pip install -r requirements.txt
cd dynamo
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/export_serde_test_results.xml --ir dynamo models/test_export_serde.py
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/reexport_test_results.xml --ir dynamo models/test_reexport.py
popd
Expand Down Expand Up @@ -224,7 +230,9 @@ jobs:
export USE_HOST_DEPS=1
export CI_BUILD=1
pushd .
cd tests/py/dynamo
cd tests/py
python -m pip install -r requirements.txt
cd dynamo
python -m pytest -ra -n 10 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_be_test_results.xml backend/
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_complete_be_e2e_test_results.xml --ir torch_compile models/test_models.py
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/torch_compile_dyn_models_export.xml --ir torch_compile models/test_dyn_models.py
Expand Down Expand Up @@ -255,7 +263,9 @@ jobs:
export USE_HOST_DEPS=1
export CI_BUILD=1
pushd .
cd tests/py/dynamo
cd tests/py
python -m pip install -r requirements.txt
cd dynamo
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_test_results.xml --ignore runtime/test_002_cudagraphs_py.py --ignore runtime/test_002_cudagraphs_cpp.py runtime/
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
Expand Down Expand Up @@ -286,7 +296,9 @@ jobs:
export USE_HOST_DEPS=1
export CI_BUILD=1
pushd .
cd tests/py/dynamo
cd tests/py
python -m pip install -r requirements.txt
cd dynamo
nvidia-smi
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_cudagraphs_cpp_test_results.xml runtime/test_002_cudagraphs_cpp.py || true
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_cudagraphs_py_test_results.xml runtime/test_002_cudagraphs_py.py || true
Expand Down

0 comments on commit 371fe76

Please sign in to comment.