File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments