Skip to content

Commit f24351a

Browse files
authored
Update mul int16 test
Differential Revision: D83437473 Pull Request resolved: #14646
1 parent 9ab5592 commit f24351a

File tree

2 files changed

+1
-7
lines changed

2 files changed

+1
-7
lines changed

backends/arm/operators/op_repeat.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def define_node(
4444
validate_valid_dtype(
4545
self.target,
4646
[inputs[0], output],
47-
[ts.DType.INT8, ts.DType.INT32, ts.DType.FP32],
47+
[ts.DType.INT8, ts.DType.INT32, ts.DType.INT16, ts.DType.FP32],
4848
output.tosa_spec,
4949
)
5050

backends/arm/test/ops/test_mul.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -338,9 +338,6 @@ def test_mul_tensor_16a8w_tosa_INT(test_data: input_t1):
338338

339339
@common.parametrize("test_data", test_data_suite)
340340
@common.XfailIfNoCorstone300
341-
@pytest.mark.xfail(
342-
reason="Vela compilation fails with 'Invalid arguments' for int16 mul operations. See: https://github.com/pytorch/executorch/issues/13947"
343-
)
344341
def test_mul_tensor_16a8w_u55_INT16(test_data: input_t1):
345342
"""Test mul operation with 16A8W quantization on U55 (16-bit activations, 8-bit weights)"""
346343
per_channel_quantization = False
@@ -365,9 +362,6 @@ def test_mul_tensor_16a8w_u55_INT16(test_data: input_t1):
365362

366363
@common.parametrize("test_data", test_data_suite)
367364
@common.XfailIfNoCorstone320
368-
@pytest.mark.xfail(
369-
reason="Vela compilation fails with 'Invalid arguments' for int16 mul operations. See: https://github.com/pytorch/executorch/issues/13947"
370-
)
371365
def test_mul_tensor_16a8w_u85_INT16(test_data: input_t1):
372366
"""Test mul operation with 16A8W quantization on U85 (16-bit activations, 8-bit weights)"""
373367
per_channel_quantization = False

0 commit comments

Comments
 (0)