Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix: ResizeBicubicGrad causes SparseConcat to choose the wrong DoCompute kernel

This commit is contained in:
yanzhenxiang2020 2023-03-01 11:23:30 +08:00
parent 0f620a4bfd
commit 31e6ae5580
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/resize_bicubic_grad.cc
View File

@ -401,7 +401,7 @@ inline void ResizeBicubicGrad(const float *input_grad, ResizerGradState &resizer
} }
template <typename T> template <typename T>
uint32_t DoCompute(CpuKernelContext &ctx) { uint32_t ResizeBicubicGradCpuKernel::DoCompute(CpuKernelContext &ctx) {
auto input0_addr = reinterpret_cast<float *>(ctx.Input(0)->GetData()); auto input0_addr = reinterpret_cast<float *>(ctx.Input(0)->GetData());
auto output_addr = reinterpret_cast<T *>(ctx.Output(0)->GetData()); auto output_addr = reinterpret_cast<T *>(ctx.Output(0)->GetData());

6
mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/resize_bicubic_grad.h
View File

@ -22,10 +22,6 @@
#include "cpu_ops_kernel.h" #include "cpu_ops_kernel.h"
namespace aicpu { namespace aicpu {
template <typename T>
uint32_t DoCompute(CpuKernelContext &ctx);
class ResizeBicubicGradCpuKernel : public CpuKernel { class ResizeBicubicGradCpuKernel : public CpuKernel {
public: public:
~ResizeBicubicGradCpuKernel() = default; ~ResizeBicubicGradCpuKernel() = default;
@ -33,6 +29,8 @@ class ResizeBicubicGradCpuKernel : public CpuKernel {
private: private:
uint32_t GetInputAndCheck(CpuKernelContext &ctx); uint32_t GetInputAndCheck(CpuKernelContext &ctx);
template <typename T>
uint32_t DoCompute(CpuKernelContext &ctx);
}; };
} // namespace aicpu } // namespace aicpu
#endif #endif