Skip to content

Commit 5c9c49d

Browse files
committed
fix: Removed sawb_new 16bin quantizers
Signed-off-by: Brandon Groth <[email protected]>
1 parent 277228b commit 5c9c49d

File tree

2 files changed

+193
-190
lines changed

2 files changed

+193
-190
lines changed

fms_mo/quant_refactor/per_channel_ste.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -373,13 +373,13 @@ def backward(ctx, grad_output):
373373
torch.FloatTensor, None,...,None: STE Gradient
374374
"""
375375
return grad_output, None, None, None, None, None, None
376-
376+
377377
class PerChannelSTESAWB(PerChannelSTE):
378378
"""
379379
PerChannelSTE Base for SAWB
380380
381381
Extends:
382-
PerChSTE
382+
PerChannelSTE
383383
"""
384384
@staticmethod
385385
def forward(

0 commit comments

Comments
 (0)