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

"polish cmake file"

上级 2573ac14
......@@ -88,7 +88,6 @@ function(op_library TARGET)
set(pybind_flag 1)
# 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_CPU_ONLY_OP(ncclInit);\n")
endif()
# reduce_op contains several operators
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册