Skip to content

Commit 5e7a100

Browse files
committed
update to share code
1 parent 0e0bee5 commit 5e7a100

File tree

6 files changed

+490
-267
lines changed

6 files changed

+490
-267
lines changed

exir/passes/_quant_patterns_and_replacements.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ def embedding_2bit(
206206
group_size = (4 * weight.size(1)) // (
207207
weight_scales.size(1) if weight_scales.dim() == 2 else 1
208208
)
209-
weight_0 = (weight & 3)
209+
weight_0 = weight & 3
210210
weight_1 = (weight & 12) >> 2
211211
weight_2 = (weight & 48) >> 4
212212
weight_3 = (weight & 192) >> 6
@@ -261,7 +261,7 @@ def embedding_2bit_dtype(
261261
group_size = (4 * weight.size(1)) // (
262262
weight_scales.size(1) if weight_scales.dim() == 2 else 1
263263
)
264-
weight_0 = (weight & 3)
264+
weight_0 = weight & 3
265265
weight_1 = (weight & 12) >> 2
266266
weight_2 = (weight & 48) >> 4
267267
weight_3 = (weight & 192) >> 6

0 commit comments

Comments
 (0)