if(WITH_GPU)
    nv_library(math_function SRCS math_function.cc math_function.cu im2col.cc DEPS cblas device_context)
else()
    cc_library(math_function SRCS math_function.cc im2col.cc DEPS cblas device_context)
endif()

nv_test(math_function_test SRCS math_function_test.cc DEPS math_function tensor)