未验证 提交 fd6a0607 编写于 作者: J jjyaoao 提交者: GitHub

change cmake/operators.cmake (#52679)

上级 a6aa701e
...@@ -62,7 +62,6 @@ function(op_library TARGET) ...@@ -62,7 +62,6 @@ function(op_library TARGET)
set(hip_cc_srcs) set(hip_cc_srcs)
set(xpu_cc_srcs) set(xpu_cc_srcs)
set(xpu_kp_cc_srcs) set(xpu_kp_cc_srcs)
set(npu_cc_srcs)
set(mlu_cc_srcs) set(mlu_cc_srcs)
set(cudnn_cu_cc_srcs) set(cudnn_cu_cc_srcs)
set(miopen_cu_cc_srcs) set(miopen_cu_cc_srcs)
...@@ -309,12 +308,7 @@ function(op_library TARGET) ...@@ -309,12 +308,7 @@ function(op_library TARGET)
if(WITH_UNITY_BUILD AND op_library_UNITY) if(WITH_UNITY_BUILD AND op_library_UNITY)
# Combine the cc source files. # Combine the cc source files.
compose_unity_target_sources( compose_unity_target_sources(
${UNITY_TARGET} ${UNITY_TARGET} cc ${cc_srcs} ${mkldnn_cc_srcs} ${xpu_cc_srcs}
cc
${cc_srcs}
${mkldnn_cc_srcs}
${xpu_cc_srcs}
${npu_cc_srcs}
${mlu_cc_srcs}) ${mlu_cc_srcs})
if(TARGET ${UNITY_TARGET}) if(TARGET ${UNITY_TARGET})
# If `UNITY_TARGET` exists, add source files to `UNITY_TARGET`. # If `UNITY_TARGET` exists, add source files to `UNITY_TARGET`.
...@@ -331,8 +325,7 @@ function(op_library TARGET) ...@@ -331,8 +325,7 @@ function(op_library TARGET)
else() else()
cc_library( cc_library(
${TARGET} ${TARGET}
SRCS ${cc_srcs} ${mkldnn_cc_srcs} ${xpu_cc_srcs} ${npu_cc_srcs} SRCS ${cc_srcs} ${mkldnn_cc_srcs} ${xpu_cc_srcs} ${mlu_cc_srcs}
${mlu_cc_srcs}
DEPS ${op_library_DEPS} ${op_common_deps}) DEPS ${op_library_DEPS} ${op_common_deps})
endif() endif()
endif() endif()
...@@ -344,7 +337,6 @@ function(op_library TARGET) ...@@ -344,7 +337,6 @@ function(op_library TARGET)
list(LENGTH mkldnn_cc_srcs mkldnn_cc_srcs_len) list(LENGTH mkldnn_cc_srcs mkldnn_cc_srcs_len)
list(LENGTH xpu_cc_srcs xpu_cc_srcs_len) list(LENGTH xpu_cc_srcs xpu_cc_srcs_len)
list(LENGTH miopen_cu_cc_srcs miopen_cu_cc_srcs_len) list(LENGTH miopen_cu_cc_srcs miopen_cu_cc_srcs_len)
list(LENGTH npu_cc_srcs npu_cc_srcs_len)
list(LENGTH mlu_cc_srcs mlu_cc_srcs_len) list(LENGTH mlu_cc_srcs mlu_cc_srcs_len)
# Define operators that don't need pybind here. # Define operators that don't need pybind here.
...@@ -567,7 +559,6 @@ function(register_operators) ...@@ -567,7 +559,6 @@ function(register_operators)
"*_op.cc") "*_op.cc")
string(REPLACE "_mkldnn" "" OPS "${OPS}") string(REPLACE "_mkldnn" "" OPS "${OPS}")
string(REPLACE "_xpu" "" OPS "${OPS}") string(REPLACE "_xpu" "" OPS "${OPS}")
string(REPLACE "_npu" "" OPS "${OPS}")
string(REPLACE "_mlu" "" OPS "${OPS}") string(REPLACE "_mlu" "" OPS "${OPS}")
string(REPLACE ".cc" "" OPS "${OPS}") string(REPLACE ".cc" "" OPS "${OPS}")
list(REMOVE_DUPLICATES OPS) list(REMOVE_DUPLICATES OPS)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册