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

!14345 [GPU] index_add op indices tensor must be 1D 

From: @tom__chen
Reviewed-by: @robingrosman,@mikef
Signed-off-by: @robingrosman
This commit is contained in:
mindspore-ci-bot 2021-03-31 01:34:10 +08:00 committed by Gitee
parent a95ae50a96 ea02bf6e7a
commit 5312cb372e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ops/operations/math_ops.py
View File

@ -4600,6 +4600,7 @@ class IndexAdd(PrimitiveWithInfer):
validator.check("x rank", len(x_shape), "y rank", len(y_shape), Rel.EQ, self.name)
x_rank = len(x_shape)
validator.check_int_range(self.axis, -x_rank - 1, x_rank, Rel.INC_NEITHER, 'axis', self.name)
validator.check_equal_int(len(idx_shape), 1, "rank of idx_shape", self.name)
validator.check("size of indices", idx_shape[0], "dimension of y[axis]", y_shape[self.axis],
Rel.EQ, self.name)
axis = self.axis if self.axis >= 0 else x_rank + self.axis