Skip to content

Commit ad5f403

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

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
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 & 1 deletion
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,7 +158,7 @@ def quantize( # noqa C901
157158
quantize_(
158159
model,
159160
Int8DynamicActivationIntxWeightConfig(
160-
weight_dtype=torch.int4,
161+
weight_dtype=torch.int4, # pyre-ignore[16]
161162
weight_granularity=PerGroup(group_size),
162163
),
163164
)

0 commit comments

Comments
 (0)