Skip to content

Commit 389a427

Browse files
committed
feat: Enabled PTnative SAWB perCh test
Signed-off-by: Brandon Groth <[email protected]>
1 parent a852ca8 commit 389a427

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/quantizers/test_sawb.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -425,7 +425,6 @@ def test_sawbnew_symmetric_perCh(
425425
tensor: torch.FloatTensor,
426426
quantizer_symmetric_perCh: dict,
427427
base_options: dict,
428-
other_options: dict, # only 1 STE for this case right now
429428
):
430429
"""
431430
Test SAWB_new w/ symmetric tensors for perCh
@@ -458,9 +457,10 @@ def test_sawbnew_symmetric_perCh(
458457
qscheme = qscheme,
459458
)
460459

461-
# Set base quantizer and SAWB options ; save nativePT
462-
native_pt = base_options["nativePT"]
463-
base_options["nativePT"] = False # Not supported for SAWB
460+
# Create only set of other options for SAWB perCh STEs
461+
other_options = {"clipSTE": True, "align_zero": True, "use16bins": False}
462+
463+
# Set base quantizer and SAWB options
464464
set_base_options(
465465
sawbnew_quantizer_symmetric_perCh, torch_quantizer_symmetric_perCh, base_options
466466
)

0 commit comments

Comments
 (0)