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

spacetobatch

This commit is contained in:
zong_shuai 2021-08-24 15:57:34 +08:00
parent 6defa8e6df
commit 9227a74ed8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/spacetobatch_impl.cu
View File

@ -66,7 +66,7 @@ void CalSpaceToBatch(const size_t size, const T *input, const size_t in,
const size_t oc, const size_t pad_up, const size_t pad_dn,
const size_t pad_lft, const size_t pad_rht, const size_t block_num,
T *output, cudaStream_t cuda_stream) {
cudaMemset(output, 0, on * oc * oh * ow * size(T));
//cudaMemset(output, 0, on * oc * oh * ow * size(T));
SpaceToBatch<<<GET_BLOCKS(size), GET_THREADS, 0, cuda_stream>>>(
size, input, in, ih, iw, ic, on, oh, ow, oc, pad_up, pad_dn, pad_lft, pad_rht, block_num, output);
return;