File tree Expand file tree Collapse file tree 6 files changed +12
-12
lines changed
Expand file tree Collapse file tree 6 files changed +12
-12
lines changed Original file line number Diff line number Diff line change 2323from fms_mo .quant_refactor .base_quant import Qscheme , Quantizer
2424from fms_mo .quant_refactor .per_tensor_ste import PerTensorSTE , PerTensorSTE_PTnative
2525
26- clip_valn_default = torch .tensor (- 8.0 )
27- clip_val_default = torch .tensor (8.0 )
26+ clip_valn_default = torch .tensor ([ - 8.0 ] )
27+ clip_val_default = torch .tensor ([ 8.0 ] )
2828qscheme_per_tensor = Qscheme (
2929 unit = "perT" ,
3030 symmetric = False ,
Original file line number Diff line number Diff line change 2323from fms_mo .quant_refactor .base_quant import Qscheme , Quantizer
2424from fms_mo .quant_refactor .per_tensor_ste import PerTensorSTE , PerTensorSTE_PTnative
2525
26- clip_valn_default = torch .tensor (- 8.0 )
27- clip_val_default = torch .tensor (8.0 )
26+ clip_valn_default = torch .tensor ([ - 8.0 ] )
27+ clip_val_default = torch .tensor ([ 8.0 ] )
2828qscheme_per_tensor = Qscheme (
2929 unit = "perT" ,
3030 symmetric = True ,
Original file line number Diff line number Diff line change 2323from fms_mo .quant_refactor .base_quant import Qscheme , Quantizer
2424from fms_mo .quant_refactor .per_tensor_ste import PerTensorSTE , PerTensorSTE_PTnative
2525
26- clip_valn_default = torch .tensor (0.0 )
27- clip_val_default = torch .tensor (8.0 )
26+ clip_valn_default = torch .tensor ([ 0.0 ] )
27+ clip_val_default = torch .tensor ([ 8.0 ] )
2828qscheme_per_tensor = Qscheme (
2929 unit = "perT" ,
3030 symmetric = False ,
Original file line number Diff line number Diff line change 2323from fms_mo .quant_refactor .base_quant import Qscheme , Quantizer
2424from fms_mo .quant_refactor .per_tensor_ste import PerTensorSTE , PerTensorSTE_PTnative
2525
26- clip_valn_default = torch .tensor (- 8.0 )
27- clip_val_default = torch .tensor (8.0 )
26+ clip_valn_default = torch .tensor ([ - 8.0 ] )
27+ clip_val_default = torch .tensor ([ 8.0 ] )
2828qscheme_per_tensor = Qscheme (
2929 unit = "perT" ,
3030 symmetric = True ,
Original file line number Diff line number Diff line change 3030 PerTensorSTEQmax_PTnative ,
3131)
3232
33- clip_valn_default = torch .FloatTensor ( - 8.0 )
34- clip_val_default = torch .FloatTensor ( 8.0 )
33+ clip_valn_default = torch .Tensor ([ - 8.0 ] )
34+ clip_val_default = torch .Tensor ([ 8.0 ] )
3535qscheme_per_tensor = Qscheme (
3636 unit = "perT" ,
3737 symmetric = False ,
Original file line number Diff line number Diff line change 3636# from fms_mo.quant_refactor.linear_utils import linear_dequantize, linear_quantize
3737from fms_mo .quant_refactor .sawb_utils import sawb_params , sawb_params_code
3838
39- clip_valn_default = torch .tensor (- 8.0 )
40- clip_val_default = torch .tensor (8.0 )
39+ clip_valn_default = torch .tensor ([ - 8.0 ] )
40+ clip_val_default = torch .tensor ([ 8.0 ] )
4141qscheme_per_tensor = Qscheme (
4242 unit = "perT" ,
4343 symmetric = False ,
You can’t perform that action at this time.
0 commit comments