未验证 提交 c665400b 编写于 作者: U umiswing 提交者: GitHub

[cutlass] Fix make (#51718)

上级 5cb95856
...@@ -38,19 +38,27 @@ ExternalProject_Add( ...@@ -38,19 +38,27 @@ ExternalProject_Add(
PREFIX ${CUTLASS_PREFIX_DIR} PREFIX ${CUTLASS_PREFIX_DIR}
UPDATE_COMMAND "" UPDATE_COMMAND ""
CONFIGURE_COMMAND "" CONFIGURE_COMMAND ""
BUILD_COMMAND BUILD_COMMAND ""
INSTALL_COMMAND ""
TEST_COMMAND "")
add_custom_target(
cutlass_codegen
COMMAND
rm -rf rm -rf
${CMAKE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/cutlass_generator/build && ${CMAKE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/cutlass_generator/build
COMMAND
mkdir -p mkdir -p
${CMAKE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/cutlass_generator/build/generated/gemm ${CMAKE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/cutlass_generator/build/generated/gemm
&& ${PYTHON_EXECUTABLE} -B COMMAND
${PYTHON_EXECUTABLE} -B
${CMAKE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/cutlass_generator/gather_gemm_scatter_generator.py ${CMAKE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/cutlass_generator/gather_gemm_scatter_generator.py
"${THIRD_PARTY_PATH}/cutlass/src/extern_cutlass/tools/library/scripts/" "${THIRD_PARTY_PATH}/cutlass/src/extern_cutlass/tools/library/scripts/"
"${CMAKE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/cutlass_generator/build" "${CMAKE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/cutlass_generator/build"
"${CMAKE_CUDA_COMPILER_VERSION}" "${CMAKE_CUDA_COMPILER_VERSION}"
INSTALL_COMMAND "" VERBATIM)
TEST_COMMAND "")
add_library(cutlass INTERFACE) add_library(cutlass INTERFACE)
add_dependencies(cutlass_codegen extern_cutlass)
add_dependencies(cutlass extern_cutlass) add_dependencies(cutlass extern_cutlass)
...@@ -174,6 +174,9 @@ set(ADD_PHI_KERNELS phi_cpu) ...@@ -174,6 +174,9 @@ set(ADD_PHI_KERNELS phi_cpu)
if(WITH_GPU OR WITH_ROCM) if(WITH_GPU OR WITH_ROCM)
if(WITH_GPU) if(WITH_GPU)
add_library(phi_gpu ${kernel_cu}) add_library(phi_gpu ${kernel_cu})
if(WITH_CUTLASS)
add_dependencies(phi_gpu cutlass_codegen)
endif()
elseif(WITH_ROCM) elseif(WITH_ROCM)
hip_add_library(phi_gpu STATIC ${kernel_cu}) hip_add_library(phi_gpu STATIC ${kernel_cu})
endif() endif()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册