CMakeLists.txt 315 字节
Newer Older
Y
Yan Chunwei 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13
if(NOT LITE_WITH_CUDA)
    return()
endif()

message(STATUS "compile with lite CUDA kernels")

nv_library(mul_compute_cuda SRCS mul_compute.cc DEPS tensor)
lite_cc_library(io_copy_compute_cuda SRCS io_copy_compute.cc DEPS tensor)

nv_library(kernels_cuda DEPS mul_compute_cuda io_copy_compute_cuda cuda_blas)