华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
问题背景:在昇腾910B2的开发机上配置好PyTorch 2.3.1和Torch-NPU库之后,编写基本的测试脚本,进行3x4和4x4的矩阵乘法,可以正常完成运算但存在Warning,如下图所示
问题描述:1.请解释出现Device do not support double dtype now, dtype cast repalce with float的原因。 2.请问在Torch-NPU下应当如何指定使用Float16数据类型,并避免产生上述Wanring。 期待您的解答,非常感谢
我要发帖子
问题背景:在昇腾910B2的开发机上配置好PyTorch 2.3.1和Torch-NPU库之后,编写基本的测试脚本,进行3x4和4x4的矩阵乘法,可以正常完成运算但存在Warning,如下图所示
问题描述:1.请解释出现Device do not support double dtype now, dtype cast repalce with float的原因。
2.请问在Torch-NPU下应当如何指定使用Float16数据类型,并避免产生上述Wanring。
期待您的解答,非常感谢