未验证 提交 cf9e433f 编写于 作者: M Masahiro Tanaka 提交者: GitHub

fix import path to op_builder (#2687)

Co-authored-by: NOlatunji Ruwase <olruwase@microsoft.com>
上级 a3d7f106
......@@ -450,7 +450,7 @@ class OpBuilder(ABC):
extra_link_args=self.strip_empty_entries(self.extra_ldflags()))
def load(self, verbose=True):
from ...git_version_info import installed_ops, torch_info
from deepspeed.git_version_info import installed_ops, torch_info
if installed_ops[self.name]:
# Ensure the op we're about to load was compiled with the same
# torch/cuda versions we are currently using at runtime.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册