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()