Enable int16 for op permute (#15256)

Summary:

Enable int16 for op permute

Reviewed By: Ninja91

Differential Revision: D84948536

Author: 3l1

backends/arm/operators/op_permute.py

@@ -117,7 +117,7 @@ def define_node(
         validate_valid_dtype(
             self.target,
             [inputs[0], output],
-            [ts.DType.INT8, ts.DType.INT32, ts.DType.FP32],
+            [ts.DType.INT8, ts.DType.INT16, ts.DType.INT32, ts.DType.FP32],
             output.tosa_spec,
         )
 

backends/arm/test/ops/test_permute.py

@@ -19,7 +19,6 @@
     TosaPipelineINT,
     VgfPipeline,
 )
-from torchvision.ops import Permute
 
 input_t1 = Tuple[torch.Tensor]  # Input x
 
@@ -42,10 +41,10 @@ class SimplePermute(torch.nn.Module):
     def __init__(self, dims: list[int]):
         super().__init__()
 
-        self.permute = Permute(dims=dims)
+        self.dims = dims
 
     def forward(self, x):
-        return self.permute(x)
+        return torch.permute(x, self.dims)
 
 
 @common.parametrize("test_data", test_data_suite)

backends/arm/test/targets.bzl

@@ -20,6 +20,7 @@ def define_arm_tests():
         "ops/test_cat.py",
         "ops/test_linear.py", 
         "ops/test_mul.py",
+        "ops/test_permute.py",
         "ops/test_slice.py",
         "ops/test_sigmoid.py",
         "ops/test_sub.py",