未验证 提交 d3cad051 编写于 作者: R Reza Yazdani 提交者: GitHub

fixing the inference build path when pre-building the inference op (#1755)

上级 56eac382
......@@ -10,7 +10,7 @@ class InferenceBuilder(CUDAOpBuilder):
super().__init__(name=name)
def absolute_name(self):
return f'deepspeed.ops.transformer_inference.{self.NAME}_op'
return f'deepspeed.ops.transformer.inference.{self.NAME}_op'
def sources(self):
return [
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册