算子不支持会退回CPU,我需要一个查询训练用到的算子,替代并强制训练在NPU上完成的方法
收藏回复举报
算子不支持会退回CPU,我需要一个查询训练用到的算子,替代并强制训练在NPU上完成的方法
t('forum.solved') 已解决
新人帖
发表于2025-07-01 16:56:54
0 查看
Warning: since the loaded file is not a zipfile, only "torch.device" and "str" type parameters are currently supported for parameter types of map_locationIf parameter types of map_location is "Callable[[torch.Tensor, str], torch.Tensor]" or "Dict[str, str]", which is only support for zipfile,all tensors are currently loaded onto the CPU, which may introduce problems 

Warning: since the loaded file is not a zipfile, only "torch.device" and "str" type parameters are currently supported for parameter types of map_locationIf parameter types of map_location is "Callable[[torch.Tensor, str], torch.Tensor]" or "Dict[str, str]", which is only support for zipfile,all tensors are currently loaded onto the CPU, which may introduce problems 

[W compiler_depend.ts:51] Warning: CAUTION: The operator 'aten::_linalg_eigh.eigenvalues' is not currently supported on the NPU backend and will fall back to run on the CPU. This may have performance implications. (function npu_cpu_fallback)

我要发帖子