forked from mindspore-Ecosystem/mindspore
!8991 Optimize log level to DEBUG in array reduce gpu kernel
From: @zpac Reviewed-by: @cristoval,@kisnwang Signed-off-by: @cristoval
This commit is contained in:
commit
38f647e122
|
@ -54,7 +54,7 @@ class ArrayReduceGpuKernel : public GpuKernel {
|
||||||
const float alpha = 1;
|
const float alpha = 1;
|
||||||
const float beta = 0;
|
const float beta = 0;
|
||||||
if (all_match_) {
|
if (all_match_) {
|
||||||
MS_LOG(WARNING)
|
MS_LOG(DEBUG)
|
||||||
<< "The corresponding dimensions of the input and output tensors all match. No need to call cuDNN kernel.";
|
<< "The corresponding dimensions of the input and output tensors all match. No need to call cuDNN kernel.";
|
||||||
CHECK_CUDA_RET_WITH_EXCEPT(cudaMemcpyAsync(output_addr, input_addr, inputs[0]->size, cudaMemcpyDeviceToDevice,
|
CHECK_CUDA_RET_WITH_EXCEPT(cudaMemcpyAsync(output_addr, input_addr, inputs[0]->size, cudaMemcpyDeviceToDevice,
|
||||||
reinterpret_cast<cudaStream_t>(stream_ptr)),
|
reinterpret_cast<cudaStream_t>(stream_ptr)),
|
||||||
|
|
Loading…
Reference in New Issue