提交 626ff3b7 编写于 作者: D Dong Zhihong

"polish cmake file"

上级 2573ac14
...@@ -88,7 +88,6 @@ function(op_library TARGET) ...@@ -88,7 +88,6 @@ function(op_library TARGET)
set(pybind_flag 1) set(pybind_flag 1)
# It's enough to just adding one operator to pybind # It's enough to just adding one operator to pybind
file(APPEND ${pybind_file} "USE_GPU_ONLY_OP(ncclAllReduce);\n") file(APPEND ${pybind_file} "USE_GPU_ONLY_OP(ncclAllReduce);\n")
file(APPEND ${pybind_file} "USE_CPU_ONLY_OP(ncclInit);\n")
endif() endif()
# reduce_op contains several operators # reduce_op contains several operators
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册