1 系统环境

硬件环境(Ascend/GPU/CPU): Ascend/GPU/CPU
MindSpore版本: 2.0.0
执行模式(PyNative/ Graph): PyNative模式
Python版本: 3.9.15
操作系统平台: Linux 18.04

2 报错信息

2.1 问题描述

mindspore.numpy.unique()不支持0 shape tensor

2.2 报错信息


RuntimeError: Malloc for kernel output failed, Memory isn't enough, node:Default/Unique-op0 ---------------------------------------------------- - C++ Call Stack: (For framework developers) ---------------------------------------------------- mindspore/ccsrc/runtime/pynative/run_op_helper.cc:473 LaunchKernels

2.3 脚本代码(代码格式,可上传附件)


a = ms.Tensor([]) ms.numpy.unique(a)

3 根因分析

在1中的系统环境下,执行上述代码不会出现上述报错,只是会出现WARNING信息,代码还是输出预期结果。
在gitee看到了关于该问题的issue,实际报错应该是:


RuntimeError: Malloc for kernel output failed, Memory isn't enough, node:Default/Unique-op0

应该是环境配置或参数配置(比如batch size等)等问题引发的,需要排查一下。
linux环境中(Python3.9,mindpsore2.0.0)代码内容:

680×143 3.63 KB

 执行结果: 

 window(Python3.9,mindpsore1.7.0)也可以正常执行: 

1174×112 6.32 KB

4 解决方案

在gitee看到了关于该问题的issue,实际报错部分应该是:


RuntimeError: Malloc for kernel output failed, Memory isn't enough, node:Default/Unique-op0

应该是环境配置或参数配置(比如batch size等)等问题引发的,需要了解更多代码进行问题原因排查。

Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐