Skip to content

Commit

Permalink
Fix typo & delete useless code
Browse files Browse the repository at this point in the history
  • Loading branch information
LLee233 committed Dec 3, 2023
1 parent 1a5f396 commit f14418d
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 2 deletions.
1 change: 0 additions & 1 deletion paddle/phi/backends/onednn/onednn_helper.h
Original file line number Diff line number Diff line change
Expand Up @@ -245,7 +245,6 @@ inline void GetSqueezeNewAxes(std::vector<int32_t>& axes) {
for (size_t i = 0; i < axes.size(); i++) {
axes[i] = axes[i] >= 0 ? perm[i] : perm[axes[i] + 4];
}
//(*attrs)["axes"] = axes;
}

inline void MatchShapeToLayout(DenseTensor* tensor_in,
Expand Down
2 changes: 1 addition & 1 deletion paddle/phi/kernels/onednn/squeeze_kernel.cc
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ void SqueezeInferKernel(const Context& dev_ctx,
auto x_dims = x.dims();
std::vector<int32_t> tmp(axes.GetData().begin(), axes.GetData().end());

// Currently there is only tranformation for tensors, while attr axes still
// Currently there is only transformation for tensors, while attr axes still
// follows default dtype instead of oneDNN dtype, so here manually change it
// TODO(Li Xinyi): Is there a way to achieve it at earlier stage?
funcs::GetSqueezeNewAxes(tmp);
Expand Down

0 comments on commit f14418d

Please sign in to comment.