Skip to content

Commit f8f6ce9

Browse files
authored
Update job_pytorch_models_tests.yml
1 parent 06cfbe2 commit f8f6ce9

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

.github/workflows/job_pytorch_models_tests.yml

+9-9
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,15 @@ jobs:
129129
TEST_DEVICE: CPU
130130
OP_REPORT_FILE: ${{ env.INSTALL_TEST_DIR }}/TEST-torch_unsupported_ops.log
131131

132+
- name: TorchFX GPTQ Pattern Test
133+
if: ${{ inputs.model_scope == 'precommit' }}
134+
run: |
135+
export PYTHONPATH=${MODEL_HUB_TESTS_INSTALL_DIR}:$PYTHONPATH
136+
python3 -m pytest ${MODEL_HUB_TESTS_INSTALL_DIR}/transformation_tests/test_gptq_torchfx_transformations.py -m precommit --html=${INSTALL_TEST_DIR}/TEST-torch_gptqpattern_tests.html --self-contained-html -v --tb=short
137+
env:
138+
TEST_DEVICE: CPU
139+
USE_SYSTEM_CACHE: False
140+
132141
- name: PyTorch Models Tests Not Timm or Torchvision
133142
if: ${{ inputs.model_scope == 'precommit' || inputs.model_scope == 'nightly_scope2' }}
134143
run: |
@@ -149,15 +158,6 @@ jobs:
149158
TEST_DEVICE: CPU
150159
USE_SYSTEM_CACHE: False
151160

152-
- name: TorchFX GPTQ Pattern Test
153-
if: ${{ inputs.model_scope == 'precommit' }}
154-
run: |
155-
export PYTHONPATH=${MODEL_HUB_TESTS_INSTALL_DIR}:$PYTHONPATH
156-
python3 -m pytest ${MODEL_HUB_TESTS_INSTALL_DIR}/transformation_tests/test_gptq_torchfx_transformations.py -m precommit --html=${INSTALL_TEST_DIR}/TEST-torch_gptqpattern_tests.html --self-contained-html -v --tb=short
157-
env:
158-
TEST_DEVICE: CPU
159-
USE_SYSTEM_CACHE: False
160-
161161
- name: RoPE Test
162162
if: ${{ inputs.model_scope == 'precommit' }}
163163
run: |

0 commit comments

Comments
 (0)