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

nv_test(math_function_test SRCS math_function_test.cc DEPS math_function tensor)
cc_test(im2col_test SRCS im2col_test.cc DEPS math_function tensor)
cc_test(pool_test_maxPool2d_test SRCS pool_test_maxPool2d.cc DEPS math_function tensor)
cc_test(pool_test_maxPool3d_test SRCS pool_test_maxPool3d.cc DEPS math_function tensor)