switch to built-in functions for artificial NaN

apache · Jun 5, 2018 · 4115730 · 4115730
1 parent 9a56996
commit 4115730
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/operator/nn/pool.cuh b/src/operator/nn/pool.cuh
@@ -323,7 +323,7 @@ __global__ void pool_sum_3d_gpu_kernel(const int nthreads, const DType* in_data,
  }
  }
  out_data[index] = (pool_size == 0) ?
- DType(0.0f / pool_size) :
+ DType(nanf("")) :
  a_root_p<DType, p>::Map(sum);
  }
 }

diff --git a/src/operator/nn/pool.h b/src/operator/nn/pool.h
@@ -341,7 +341,7 @@ inline void pool_sum_3d_cpu(const DType* in_data, const TShape& ishape, const TS
  }
  }
  out_data[(pd*pooled_height+ph)*pooled_width+pw] = (pool_size == 0) ?
- DType(0.0f / pool_size) :
+ DType(nanf("")) :
  a_root_p<DType, p>::Map(sum);
  }
  }