Skip to content

Commit

Permalink
comment out non scheduled unet test
Browse files Browse the repository at this point in the history
  • Loading branch information
saienduri authored Apr 24, 2024
1 parent eb72dd0 commit 74679e3
Showing 1 changed file with 14 additions and 14 deletions.
28 changes: 14 additions & 14 deletions .github/workflows/test_iree.yml
Original file line number Diff line number Diff line change
Expand Up @@ -100,20 +100,20 @@ jobs:
run: |
source ${VENV_DIR}/bin/activate
python3.11 iree_tests/download_remote_files.py --root-dir pytorch/models
- name: "Running real weight model tests ROCm AMDGPU"
env:
IREE_TEST_CONFIG_FILES: iree_tests/configs/config_gpu_rocm_models.json
run: |
source ${VENV_DIR}/bin/activate
pytest iree_tests/pytorch/models \
-rpfE \
-k real_weights \
--no-skip-tests-missing-files \
--capture=no \
--timeout=1200 \
--retries 2 \
--retry-delay 5 \
--durations=0
# - name: "Running real weight model tests ROCm AMDGPU"
# env:
# IREE_TEST_CONFIG_FILES: iree_tests/configs/config_gpu_rocm_models.json
# run: |
# source ${VENV_DIR}/bin/activate
# pytest iree_tests/pytorch/models \
# -rpfE \
# -k real_weights \
# --no-skip-tests-missing-files \
# --capture=no \
# --timeout=1200 \
# --retries 2 \
# --retry-delay 5 \
# --durations=0
- name: "Running scheduled unet model tests ROCm AMDGPU"
env:
IREE_TEST_CONFIG_FILES: iree_tests/configs/config_sdxl_scheduled_unet_gpu_rocm.json
Expand Down

0 comments on commit 74679e3

Please sign in to comment.