wehub-resource-sync
34 lines
796 B
CMake
34 lines
796 B
CMake
add_subdirectory(eigen)
|
|
add_subdirectory(blas)
|
|
add_subdirectory(lapack)
|
|
if(WITH_MAGMA)
|
|
add_subdirectory(magma)
|
|
endif()
|
|
add_subdirectory(detail)
|
|
add_subdirectory(jit)
|
|
add_subdirectory(math)
|
|
|
|
file(
|
|
GLOB func_cc_srcs
|
|
RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
|
|
"*.cc")
|
|
if(WITH_GPU OR WITH_ROCM)
|
|
file(
|
|
GLOB func_cu_srcs
|
|
RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
|
|
"*.cu")
|
|
endif()
|
|
|
|
# Note(qili93): remove kernels not supported on DCU yet
|
|
if(WITH_ROCM)
|
|
list(REMOVE_ITEM func_cu_srcs "weight_only_gemv.cu")
|
|
endif()
|
|
|
|
# Remove fast_ln_v2 kernels on HIP and Windows platforms
|
|
if(WITH_ROCM OR WIN32)
|
|
list(REMOVE_ITEM func_cu_srcs "fast_ln_v2_bwd_kernel.cu")
|
|
list(REMOVE_ITEM func_cu_srcs "fast_ln_v2_fwd_kernel.cu")
|
|
endif()
|
|
|
|
collect_srcs(kernels_srcs SRCS ${func_cc_srcs} ${func_cu_srcs})
|