search

DeepLink-org / deeplink.framework

BSD 3-Clause "New" or "Revised" License
55 stars 28 forks source link

feat(dipu): pass through Torch_VERSION to DIOPI #881

Closed wiryls closed 2 months ago

wiryls commented 2 months ago

DIOPI 在 Ascend 有一段逻辑是 import torch 然后获取对应版本。使用 pip install ./dipu 时，环境中不包含 torch，这里会报错，因此选择从 DIPU 直接传递该数至 DIOPI。

关联这个提交 https://github.com/DeepLink-org/DIOPI/pull/1295

wiryls commented 2 months ago

由于 DIOPI 变化了获取版本的方式，这里不再需要传入 Torch_VERSION，因此关闭 PR