Skip to content

Commit 68f3cda

Browse files
metascroyfacebook-github-bot
authored andcommitted
pyre-fix (#14241)
Summary: Forward fix for pyre errors in D82242003 Differential Revision: D82265586
1 parent b23f883 commit 68f3cda

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

backends/xnnpack/test/ops/test_linear.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -395,7 +395,9 @@ def _test_groupwise_dq_linear(
395395
quantize_(
396396
mod,
397397
Int8DynamicActivationIntxWeightConfig(
398-
weight_dtype=torch.int4, weight_granularity=PerGroup(group_size)
398+
# pyre-ignore[16]
399+
weight_dtype=torch.int4,
400+
weight_granularity=PerGroup(group_size),
399401
),
400402
)
401403
unwrap_tensor_subclass(mod)

examples/models/llama/source_transformation/quantize.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,6 +135,7 @@ def quantize( # noqa C901
135135
PerAxis(0) if group_size == 0 else PerGroup(group_size)
136136
),
137137
weight_mapping_type=MappingType.SYMMETRIC,
138+
# pyre-ignore[6]
138139
intx_packing_format="opaque_torchao_auto",
139140
),
140141
)
@@ -157,6 +158,7 @@ def quantize( # noqa C901
157158
quantize_(
158159
model,
159160
Int8DynamicActivationIntxWeightConfig(
161+
# pyre-ignore[16]
160162
weight_dtype=torch.int4,
161163
weight_granularity=PerGroup(group_size),
162164
),

0 commit comments

Comments
 (0)