体验新版 GitCode,发现更多精彩内容 >>
if (WITH_GPU) nv_library(math_function SRCS math_function.cc math_function.cu DEPS cblas device_context) else() cc_library(math_function SRCS math_function.cc DEPS cblas device_context) endif()