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

!2225 GPU shape check 

Merge pull request !2225 from VectorSL/shape-check
This commit is contained in:
mindspore-ci-bot 2020-06-17 20:26:54 +08:00 committed by Gitee
parent f27de896a7 9f0ae32d15
commit f96dcb026a
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/kernel/gpu/gpu_kernel.h
View File

@ -64,6 +64,9 @@ class GpuKernel : public KernelMod {
}
// expand Nd Shape to 4d (N in [0,4])
void ShapeNdTo4d(const std::vector<size_t> &src, std::vector<int> *dst) {
if (src.size() > 4) {
MS_EXCEPTION(ValueError) << src.size() << "-D data is not supported!";
}
dst->push_back(src.size() < 4 ? 1 : SizeToInt(src[src.size() - 4]));
dst->push_back(src.size() < 3 ? 1 : SizeToInt(src[src.size() - 3]));
dst->push_back(src.size() < 2 ? 1 : SizeToInt(src[src.size() - 2]));