Skip to content

Commit 45dc0f1

Browse files
committed
Update on "Add 16A8W linear ops support and test"
- Adds linear ops test using the 16A8W config in INT16 profile. - Adds support in view ops validation for INT16 Dtype. - Validated with TOSA pipeline test. - Checked earlier marked flaky tests no longer flaky and remove markers. Note: Not verified with tosa reference model run. bypass-github-export-checks bypass-github-pytorch-ci-checks bypass-github-executorch-ci-checks Differential Revision: [D81550511](https://our.internmc.facebook.com/intern/diff/D81550511/) [ghstack-poisoned]
2 parents ee4583a + 0323eb0 commit 45dc0f1

File tree

2 files changed

+15
-15
lines changed

2 files changed

+15
-15
lines changed

backends/arm/operators/op_transpose.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -44,17 +44,17 @@ def define_node(
4444

4545
validate_num_inputs(self.target, inputs, 2)
4646
validate_same_dtype(self.target, [inputs[0], output], ts)
47-
48-
valid_dtypes = []
49-
if self.tosa_spec.support_integer():
50-
valid_dtypes.extend([ts.DType.INT8, ts.DType.INT16, ts.DType.INT32])
51-
if self.tosa_spec.support_float():
52-
valid_dtypes.extend([ts.DType.FP32])
53-
5447
validate_valid_dtype(
5548
self.target,
5649
[inputs[0], output],
57-
valid_dtypes,
50+
[
51+
ts.DType.INT8,
52+
ts.DType.INT16,
53+
ts.DType.INT32,
54+
ts.DType.FP32,
55+
ts.DType.BOOL,
56+
ts.DType.FP16,
57+
],
5858
output.tosa_spec,
5959
)
6060

backends/arm/operators/op_view.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -41,16 +41,16 @@ def define_node(
4141

4242
validate_num_inputs(self.target, inputs, 2)
4343
validate_same_dtype(self.target, [inputs[0], output], ts)
44-
valid_dtypes = [ts.DType.BOOL]
45-
if self.tosa_spec.support_integer():
46-
valid_dtypes.extend([ts.DType.INT8, ts.DType.INT16, ts.DType.INT32])
47-
if self.tosa_spec.support_float():
48-
valid_dtypes.extend([ts.DType.FP32])
49-
5044
validate_valid_dtype(
5145
self.target,
5246
[inputs[0], output],
53-
valid_dtypes,
47+
[
48+
ts.DType.INT8,
49+
ts.DType.INT16,
50+
ts.DType.INT32,
51+
ts.DType.FP32,
52+
ts.DType.BOOL,
53+
],
5454
output.tosa_spec,
5555
)
5656

0 commit comments

Comments
 (0)