Skip to content

Commit 5d5cbe2

Browse files
committed
fix: Changed default clips to be 1-d tensors
Signed-off-by: Brandon Groth <[email protected]>
1 parent 7c87e01 commit 5d5cbe2

File tree

6 files changed

+12
-12
lines changed

6 files changed

+12
-12
lines changed

fms_mo/quant_refactor/pact2_new.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@
2323
from fms_mo.quant_refactor.base_quant import Qscheme, Quantizer
2424
from 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])
2828
qscheme_per_tensor = Qscheme(
2929
unit="perT",
3030
symmetric=False,

fms_mo/quant_refactor/pact2sym_new.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@
2323
from fms_mo.quant_refactor.base_quant import Qscheme, Quantizer
2424
from 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])
2828
qscheme_per_tensor = Qscheme(
2929
unit="perT",
3030
symmetric=True,

fms_mo/quant_refactor/pact_new.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@
2323
from fms_mo.quant_refactor.base_quant import Qscheme, Quantizer
2424
from 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])
2828
qscheme_per_tensor = Qscheme(
2929
unit="perT",
3030
symmetric=False,

fms_mo/quant_refactor/pactplussym_new.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@
2323
from fms_mo.quant_refactor.base_quant import Qscheme, Quantizer
2424
from 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])
2828
qscheme_per_tensor = Qscheme(
2929
unit="perT",
3030
symmetric=True,

fms_mo/quant_refactor/qmax_new.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@
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])
3535
qscheme_per_tensor = Qscheme(
3636
unit="perT",
3737
symmetric=False,

fms_mo/quant_refactor/sawb_new.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,8 @@
3636
# from fms_mo.quant_refactor.linear_utils import linear_dequantize, linear_quantize
3737
from 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])
4141
qscheme_per_tensor = Qscheme(
4242
unit="perT",
4343
symmetric=False,

0 commit comments

Comments
 (0)