Skip to content

Commit 7fc43a8

Browse files
committed
Update on "[xnnpack] Improve qd8-f16 testing"
* Added a few qd8 fp16 tests for qc4w and qc8w * Cleaned up and split fp16 and fp32 for basic tests to avoid getting confused regarding atol issues * Created a TODO to look into atol values Differential Revision: [D68350952](https://our.internmc.facebook.com/intern/diff/D68350952/) [ghstack-poisoned]
2 parents db077e5 + 8635238 commit 7fc43a8

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

backends/xnnpack/test/ops/test_linear.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -584,6 +584,7 @@ def _test_qd8_per_token_weight_per_channel_group_int4(
584584
for use_bias in [True, False]:
585585
for M, K, bl, N in zip(M_sizes, K_sizes, bl_sizes, N_sizes):
586586
lin_mod = BaseLinear(
587+
in_size=M,
587588
input_channels=K,
588589
output_channels=N,
589590
dtype=dtype,

0 commit comments

Comments
 (0)