File tree 1 file changed +4
-3
lines changed
backends/cadence/fusion_g3/operators
1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -89,6 +89,7 @@ Tensor& permute_copy_out(
89
89
90
90
int inp_shape[kTensorDimensionLimit ];
91
91
int out_shape[kTensorDimensionLimit ];
92
+ int permute_vec[kTensorDimensionLimit ];
92
93
93
94
/* input shapes and output shapes */
94
95
for (auto i = 0 ; i < in_size.size (); i++) {
@@ -99,7 +100,6 @@ Tensor& permute_copy_out(
99
100
out_shape[i] = out_size[i];
100
101
}
101
102
102
- int permute_vec[in.dim ()];
103
103
for (int i = 0 ; i < in.dim (); i++) {
104
104
permute_vec[i] = (int )dims[i];
105
105
}
@@ -112,7 +112,8 @@ Tensor& permute_copy_out(
112
112
(out.scalar_type () == ScalarType::Char) ||
113
113
(out.scalar_type () == ScalarType::UInt32 ) ||
114
114
(out.scalar_type () == ScalarType::UInt16 ) ||
115
- (out.scalar_type () == ScalarType::Byte )) &&
115
+ (out.scalar_type () == ScalarType::Byte ) ||
116
+ (out.scalar_type () == ScalarType::Float)) &&
116
117
(in.dim () <= 5 )) {
117
118
XT_KERNEL_CHECK (
118
119
ctx,
@@ -156,4 +157,4 @@ Tensor& permute_copy_out(
156
157
} // namespace native
157
158
} // namespace G3
158
159
} // namespace impl
159
- } // namespace cadence
160
+ } // namespace cadence
You can’t perform that action at this time.
0 commit comments