Skip to content

Commit d596cd7

Browse files
pytorchbotSS-JIA
andauthored
[ET-VK][ez] Test specific sizes of linear sizes in generated operator tests (#7672)
Pull Request resolved: #7667 ## Context Recent changes related to checking SPIR-V capability support at runtime have made it possible to test the 8-bit quantized linear compute shader on Android devices. Previously the test would be automatically skipped since the operator potentially uses 8-bit data types. To make the generated tests more useful, instead test real sizes of linear layer settings found in a sample model in the 8-bit linear test case. ghstack-source-id: 261524380 @exported-using-ghexport Differential Revision: [D68192068](https://our.internmc.facebook.com/intern/diff/D68192068/) Co-authored-by: Stephen Jia <[email protected]>
1 parent a18f6e8 commit d596cd7

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

backends/vulkan/test/op_tests/cases.py

+7-1
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,13 @@ def get_linear_test_suites():
169169

170170
@register_test_suite("aten._weight_int8pack_mm.default")
171171
def get_weight_int8pack_mm_inputs():
172-
MKN_list = common_MKN_list
172+
MKN_list = [
173+
[6, 480, 256],
174+
[6, 256, 1024],
175+
[6, 1024, 256],
176+
[6, 256, 256],
177+
[6, 256, 512],
178+
]
173179

174180
inputs_list = [((M, K), (N, K), (N)) for M, K, N in MKN_list]
175181

0 commit comments

Comments
 (0)