You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mssa-2021-005.patch 1.0 kB

1234567891011121314151617181920212223242526
  1. From 5aab6599e7280d2512a87434c174f13a0a2e7008 Mon Sep 17 00:00:00 2001
  2. From: lzk <liuzhongkai2@huawei.com>
  3. Date: Fri, 21 May 2021 01:25:06 -0700
  4. Subject: [PATCH] array cross the border
  5. ---
  6. .../kernel_compiler/cpu/nnacl/infer/sparse_to_dense_infer.c | 3 +++
  7. 1 file changed, 3 insertions(+)
  8. diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/sparse_to_dense_infer.c b/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/sparse_to_dense_infer.c
  9. index 4b44ec7568..89620c9634 100644
  10. --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/sparse_to_dense_infer.c
  11. +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/sparse_to_dense_infer.c
  12. @@ -27,6 +27,9 @@ int SparseToDenseInferShape(const TensorC *const *inputs, size_t inputs_size, Te
  13. #endif
  14. TensorC *output = outputs[0];
  15. + if (inputs_size < 3) {
  16. + return NNACL_INPUT_TENSOR_ERROR;
  17. + }
  18. const TensorC *input1 = inputs[1];
  19. const TensorC *input2 = inputs[2];
  20. SetDataTypeFormat(output, input2);
  21. --
  22. 2.17.1