已合并
bugfix: eliminate redundant operations in fft. #5178
yvjc创建于 6月13日
bugfix: eliminate redundant operations in fft. #5178
已合并
共 1 个文件变更+0-1
| @@ -186,7 +186,6 @@ static at::Tensor& _exec_fft(at::Tensor& out_, const at::Tensor& self_, at::IntA | |||
| 186 | [&](int64_t d) {return is_transformed_dim[d]; }); | 186 | [&](int64_t d) {return is_transformed_dim[d]; }); |
| 187 | auto self_strides = self.strides(); | 187 | auto self_strides = self.strides(); |
| 188 | at::DimVector sorted_dims = _sort_dims(self, dim, mode_code); | 188 | at::DimVector sorted_dims = _sort_dims(self, dim, mode_code); |
| 189 | - sorted_dims = _sort_dims(self, dim, mode_code); | ||
| 190 | std::copy(sorted_dims.begin(), sorted_dims.end(), dim_permute.begin()); | 189 | std::copy(sorted_dims.begin(), sorted_dims.end(), dim_permute.begin()); |
| 191 | std::sort(batch_end, dim_permute.end(), | 190 | std::sort(batch_end, dim_permute.end(), |
| 192 | [&](int64_t a, int64_t b) { return self_strides[a] > self_strides[b]; }); | 191 | [&](int64_t a, int64_t b) { return self_strides[a] > self_strides[b]; }); |