昇腾310推理卡上跑MindSpore.ops中的算子,执行失败,同样的代码在昇腾910上执行没有问题。
算子代码示例如下:
x = Tensor(np.array([[[[0.1], [0.2]],
[[0.3], [0.4]]]]), mindspore.float32)
lrn = ops.LRN()
output = lrn(x)
print(output)
1、在昇腾310推理卡上执行MindSpore.ops中的算子
2、执行报错,具体信息如下:
[ERROR] DEVICE(24646,7ff64f3a5700,python3.7):2021-11-27-16:18:03.663.456 [mindspore/ccsrc/runtime/device/ascend/ascend_memory_manager.cc:40] GetDeviceHBMSize] Get Device HBM memory size failed, ret = 0, total = 0
Traceback (most recent call last):
File "/home/autoai/scheduler/schedule.py", line 78, in schedule
msg = run_ai_workload(func, conf_dir, lib_dir, args, info)
File "/home/autoai/scheduler/schedule.py", line 102, in run_ai_workload
func(conf_dir, lib_dir, args)
File "/home/autoai/wrapper/workloads/activation/activation.py", line 46, in activation
start_time, end_time, transaction_count = execute_tasks(funcs)
File "/home/autoai/wrapper/wrapper_base.py", line 38, in wrapper
func(*args, **kwargs)
File "/home/autoai/wrapper/wrapper_base.py", line 68, in execute_tasks
func['method'](*func['args'])
File "/opt/Python-3.7.5/lib/python3.7/site-packages/mindspore/ops/primitive.py", line 247, in __call__
return _run_op(self, self.name, args)
File "/opt/Python-3.7.5/lib/python3.7/site-packages/mindspore/common/api.py", line 75, in wrapper
results = fn(*arg, **kwargs)
File "/opt/Python-3.7.5/lib/python3.7/site-packages/mindspore/ops/primitive.py", line 649, in _run_op
output = real_run_op(obj, op_name, args)
RuntimeError: mindspore/ccsrc/runtime/device/ascend/ascend_memory_manager.cc:40 GetDeviceHBMSize] Get Device HBM memory size failed, ret = 0, total = 0
*******************************************************************************************************************
该算子在310确实是不支持。