Skip to content

Commit f622f4c

Browse files
3l1facebook-github-bot
authored andcommitted
Update mul int16 test for Ethos (#14646)
Summary: bypass-github-export-checks bypass-github-pytorch-ci-checks bypass-github-executorch-ci-checks Reviewed By: digantdesai Differential Revision: D83437473
1 parent 0cd8256 commit f622f4c

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)