×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

MLU370上 v24.04-torch2.1.0-catch1.20.0-ubuntu22.04-py310 使用ComplexFloat 反向传播报错,是不支持吗? yishen2025-09-04 17:15:35 回复 1 查看 技术答疑 使用求助 经验交流
MLU370上 v24.04-torch2.1.0-catch1.20.0-ubuntu22.04-py310 使用ComplexFloat 反向传播报错,是不支持吗?
分享到:

【寒武纪硬件产品型号】:MLU370


【使用操作系统】:ubuntu


【使用驱动版本】:v5.10.31

【出错信息】:

[2025-9-4 17:9:34] [CNNL] [Warning]:When calculating multiplication of complex_float data, it is required to use [cnnlGetOpTensorWorkspaceSize_v2] to apply for workspace.

Traceback (most recent call last):

  File "/data/user/RS_IflySFUCoVideo2.0-AudioProcess/DPCodec_mlu/train_dpcodec.py", line 338, in <module>

    trainer.start_train()

  File "/data/user/RS_IflySFUCoVideo2.0-AudioProcess/DPCodec_mlu/train_dpcodec.py", line 323, in start_train

    self.__train()

  File "/data/user/RS_IflySFUCoVideo2.0-AudioProcess/DPCodec_mlu/train_dpcodec.py", line 185, in __train

    self.train_step_apollo(item)

  File "/data/user/RS_IflySFUCoVideo2.0-AudioProcess/DPCodec_mlu/train_dpcodec.py", line 214, in train_step_apollo

    loss_g.backward()

  File "/torch/venv3/pytorch/lib/python3.10/site-packages/torch/_tensor.py", line 495, in backward

    torch.autograd.backward(

  File "/torch/venv3/pytorch/lib/python3.10/site-packages/torch/autograd/__init__.py", line 251, in backward

    Variable._execution_engine.run_backward(  # Calls into the C++ engine to run the backward pass

RuntimeError: Can't get a promote dtype when common dtype is ComplexFloat


版权所有 © 2025 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭