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

[cutlass] Fix make (#51718)

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