Skip to content

Commit 600a770

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 ffd27ca commit 600a770

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
@@ -342,9 +342,6 @@ def test_mul_tensor_16a8w_tosa_INT(test_data: input_t1):
342342

343343
@common.parametrize("test_data", test_data_suite)
344344
@common.XfailIfNoCorstone300
345-
@pytest.mark.xfail(
346-
reason="Vela compilation fails with 'Invalid arguments' for int16 mul operations. See: https://github.com/pytorch/executorch/issues/13947"
347-
)
348345
def test_mul_tensor_16a8w_u55_INT16(test_data: input_t1):
349346
"""Test mul operation with 16A8W quantization on U55 (16-bit activations, 8-bit weights)"""
350347
per_channel_quantization = False
@@ -370,9 +367,6 @@ def test_mul_tensor_16a8w_u55_INT16(test_data: input_t1):
370367

371368
@common.parametrize("test_data", test_data_suite)
372369
@common.XfailIfNoCorstone320
373-
@pytest.mark.xfail(
374-
reason="Vela compilation fails with 'Invalid arguments' for int16 mul operations. See: https://github.com/pytorch/executorch/issues/13947"
375-
)
376370
def test_mul_tensor_16a8w_u85_INT16(test_data: input_t1):
377371
"""Test mul operation with 16A8W quantization on U85 (16-bit activations, 8-bit weights)"""
378372
per_channel_quantization = False

0 commit comments

Comments
 (0)