75 Star 579 Fork 1.1K

Ascend/pytorch

torch和torch-npu升级到2.2.0以上版本,报 AssertionError: Device npu not supported

DONE
需求
创建于  
2024-07-23 10:37

torch和torch-npu升级到2.2.0以上版本,报
File "/root/miniconda3/envs/qwen1.5/lib/python3.10/site-packages/torch/_inductor/graph.py", line 993, in init_wrapper_code assert wrapper_code_gen_cls is not None, f"Device {device_type} not supported" torch._dynamo.exc.BackendCompilerFailed: backend='inductor' raised: AssertionError: Device npu not supported

评论 (1)

gaoxun123 创建了需求 10个月前

当前npu不支持inductor后端,dynamo可以参考这个使用 https://gitee.com/ascend/torchair/blob/master/README.md

huangyunlong 任务状态TODO 修改为DONE 10个月前

登录 后才可以发表评论

状态
负责人
项目
里程碑
Pull Requests
关联的 Pull Requests 被合并后可能会关闭此 issue
分支
开始日期   -   截止日期
-
置顶选项
优先级
预计工期 (小时)
参与者(2)
huangyunlong-huangyunlong2022 gaoxun123-gaoxun123
Python
1
https://gitee.com/ascend/pytorch.git
git@gitee.com:ascend/pytorch.git
ascend
pytorch
pytorch

搜索帮助