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

!24257 erase warning 

Merge pull request !24257 from zong_shuai/erase_warning
This commit is contained in:
i-robot 2021-09-27 13:57:24 +00:00 committed by Gitee
parent bf74e7ce44 417096ddcc
commit e81476d1b0
3 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/extract_image_patches_gpu_kernel.h
View File

@ -49,7 +49,7 @@ class ExtractImagePatchesKernel : public GpuKernel {
size_t *t_output_shape = GetDeviceAddress<size_t>(workspace, 4);
size_t *t_output_to_nchw_axis = GetDeviceAddress<size_t>(workspace, 5);
size_t shape_size = 4 * sizeof(size_t);
const size_t shape_size = 4 * sizeof(size_t);
std::vector<size_t> to_nhwc_axis = {0, 2, 3, 1};
std::vector<size_t> to_nchw_axis = {0, 3, 1, 2};

6
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/local_response_norm_gpu_kernel.h
View File

@ -51,7 +51,7 @@ class LocalResponseNormGpuKernel : public GpuKernel {
if (use_native_) {
std::vector<size_t> to_nhwc_axis = {0, 2, 3, 1};
std::vector<size_t> to_nchw_axis = {0, 3, 1, 2};
size_t shape_size = 4 * sizeof(size_t);
const size_t shape_size = 4 * sizeof(size_t);
size_t *ws_input_shape = GetDeviceAddress<size_t>(workspace, 0);
size_t *ws_transpose_shape = GetDeviceAddress<size_t>(workspace, 1);
size_t *ws_to_nhwc_axis = GetDeviceAddress<size_t>(workspace, 2);
@ -107,7 +107,7 @@ class LocalResponseNormGpuKernel : public GpuKernel {
beta_ = GetAttr<float>(kernel_node, "beta");
use_native_ = false;
unsigned int lrnN = 2 * depth_radius_ + 1;
const unsigned int lrnN = 2 * depth_radius_ + 1;
double lrnAlpha = lrnN * alpha_;
if (lrnN < CUDNN_LRN_MIN_N || lrnN > CUDNN_LRN_MAX_N || bias_ < CUDNN_LRN_MIN_K || beta_ < CUDNN_LRN_MIN_BETA) {
use_native_ = true;
@ -191,7 +191,7 @@ class LocalResponseNormGpuKernel : public GpuKernel {
if (use_native_) {
input_size_ = num_elements_ * sizeof(T);
output_size_ = num_elements_ * sizeof(T);
size_t shape_size = 4 * sizeof(size_t);
const size_t shape_size = 4 * sizeof(size_t);
workspace_size_list_.push_back(shape_size);
workspace_size_list_.push_back(shape_size);
workspace_size_list_.push_back(shape_size);

6
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/local_response_norm_grad_gpu_kernel.h
View File

@ -55,7 +55,7 @@ class LocalResponseNormGradGpuKernel : public GpuKernel {
MS_LOG(WARNING) << "TOM: num_elements_ " << num_elements_;
std::vector<size_t> to_nhwc_axis = {0, 2, 3, 1};
std::vector<size_t> to_nchw_axis = {0, 3, 1, 2};
size_t shape_size = 4 * sizeof(size_t);
const size_t shape_size = 4 * sizeof(size_t);
size_t *ws_input_shape = GetDeviceAddress<size_t>(workspace, 0);
size_t *ws_transpose_shape = GetDeviceAddress<size_t>(workspace, 1);
size_t *ws_to_nhwc_axis = GetDeviceAddress<size_t>(workspace, 2);
@ -117,7 +117,7 @@ class LocalResponseNormGradGpuKernel : public GpuKernel {
beta_ = GetAttr<float>(kernel_node, "beta");
use_native_ = false;
unsigned int lrnN = 2 * depth_radius_ + 1;
const unsigned int lrnN = 2 * depth_radius_ + 1;
double lrnAlpha = lrnN * alpha_;
if (lrnN < CUDNN_LRN_MIN_N || lrnN > CUDNN_LRN_MAX_N || bias_ < CUDNN_LRN_MIN_K || beta_ < CUDNN_LRN_MIN_BETA) {
use_native_ = true;
@ -207,7 +207,7 @@ class LocalResponseNormGradGpuKernel : public GpuKernel {
if (use_native_) {
input_size_ = num_elements_ * sizeof(T);
output_size_ = num_elements_ * sizeof(T);
size_t shape_size = 4 * sizeof(size_t);
const size_t shape_size = 4 * sizeof(size_t);
workspace_size_list_.push_back(shape_size);
workspace_size_list_.push_back(shape_size);
workspace_size_list_.push_back(shape_size);