Skip to content

Commit a1d92d5

Browse files
committed
rm use_avx2 from pipeline args
1 parent 5eebb1f commit a1d92d5

File tree

2 files changed

+0
-5
lines changed

2 files changed

+0
-5
lines changed

tests/post_training/experimental/sparsify_activations/test_sparsify_activations_conformance.py

-1
Original file line numberDiff line numberDiff line change
@@ -78,5 +78,4 @@ def test_sparsify_activations(
7878
capsys,
7979
extra_columns,
8080
False, # memory_monitor is not used in SA
81-
None, # use_avx2 is not used in SA
8281
)

tests/post_training/test_quantize_conformance.py

-4
Original file line numberDiff line numberDiff line change
@@ -377,7 +377,6 @@ def run_pipeline(
377377
capsys: pytest.CaptureFixture,
378378
extra_columns: bool,
379379
memory_monitor: bool,
380-
use_avx2: Optional[bool] = None,
381380
):
382381
pipeline, exception_report, test_model_param = None, None, None
383382
start_time = time.perf_counter()
@@ -405,8 +404,6 @@ def run_pipeline(
405404
"memory_monitor": memory_monitor,
406405
}
407406
)
408-
if use_avx2 is not None:
409-
pipeline_kwargs["use_avx2"] = use_avx2
410407
pipeline: BaseTestPipeline = pipeline_cls(**pipeline_kwargs)
411408
try:
412409
pipeline.run()
@@ -512,5 +509,4 @@ def test_weight_compression(
512509
capsys,
513510
extra_columns,
514511
memory_monitor,
515-
use_avx2,
516512
)

0 commit comments

Comments
 (0)