mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix CPU power op

This commit is contained in:
fanjibin 2021-07-14 11:36:15 +08:00
parent 45c76beda7
commit 222777cbc2
1 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.cc
View File

@ -211,14 +211,6 @@ void ArithmeticCPUKernel<T>::FloorMod(const T *input1, const T *input2, T *out)
template <typename T>
void ArithmeticCPUKernel<T>::Pow(const T *input1, const T *input2, T *out) {
if constexpr (std::is_same_v<T, float>) {
bool broadcast = (input_shape1_ == input_shape2_) ? false : true;
int len = SizeToInt(output_size_);
float scale = 1.0;
float shift = 0.0;
Power(input1, input2, out, len, scale, shift, broadcast);
return;
}
BroadcastIterator base_iter(input_shape1_, input_shape2_, output_shape_);
if (output_size_ > MAX_POW_SERIAL_SIZE) {
auto task = [&](size_t start, size_t end) {