Skip to content

Commit 6f37e09

Browse files
authored
[Torch FX] Add Data Free AWQ Conformance Test for Torch FX (#3622)
### Changes Include Torch FX backend for data free AWQ weights compression test. ### Related tickets 169362 ### Tests Category | Job | Status | Job Number | Notes -- | -- | -- | -- | -- GH WC | Weights Compression | Pass | <a href="https://github.com/openvinotoolkit/nncf/actions/runs/16975099235">140</a> |
1 parent 04a4439 commit 6f37e09

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

tests/post_training/data/wc_reference_data.yaml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,10 @@ tinyllama_data_free_awq_backend_TORCH:
126126
metric_value: 0.85466
127127
num_int4: 94
128128
num_int8: 124
129+
tinyllama_data_free_awq_backend_FX_TORCH:
130+
metric_value: 0.85466
131+
num_int4: 94
132+
num_int8: 124
129133
tinyllama_data_free_awq_backend_ONNX:
130134
metric_value: 0.8597
131135
num_int4: 230

tests/post_training/model_scope.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -578,8 +578,7 @@
578578
awq_params=AdvancedAWQParameters(prefer_data_aware_scaling=False)
579579
),
580580
},
581-
# TODO: (andreyanufr) add torch.fx backend
582-
"backends": [BackendType.OV, BackendType.TORCH, BackendType.ONNX],
581+
"backends": [BackendType.OV, BackendType.TORCH, BackendType.FX_TORCH, BackendType.ONNX],
583582
},
584583
]
585584

0 commit comments

Comments
 (0)