Torch-NPU 使用 Float16 存在Warning
收藏回复举报
Torch-NPU 使用 Float16 存在Warning
t('forum.solved') 已解决
发表于2024-12-10 20:16:03
0 查看

问题背景:在昇腾910B2的开发机上配置好PyTorch 2.3.1和Torch-NPU库之后,编写基本的测试脚本,进行3x4和4x4的矩阵乘法,可以正常完成运算但存在Warning,如下图所示

cke_7114.png

问题描述:1.请解释出现Device do not support double dtype now, dtype cast repalce with float的原因。
2.请问在Torch-NPU下应当如何指定使用Float16数据类型,并避免产生上述Wanring。

期待您的解答,非常感谢

我要发帖子