with cutlass download

PaddlePaddle · Apr 24, 2024 · 008e268 · 008e268
1 parent f50fbdc
commit 008e268
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/paddle/phi/kernels/fusion/cutlass/gemm_epilogue/CMakeLists.txt b/paddle/phi/kernels/fusion/cutlass/gemm_epilogue/CMakeLists.txt
@@ -14,9 +14,8 @@ if(NOT DEFINED COMPUTE_CAPABILITY)
   )
 endif()
 
-include_directories("${CMAKE_CURRENT_SOURCE_DIR}/../conv2d/cutlass/include")
-include_directories(
-  "${CMAKE_CURRENT_SOURCE_DIR}/../conv2d/cutlass/tools/util/include")
+include_directories("${CMAKE_CURRENT_SOURCE_DIR}/cutlass/include")
+include_directories("${CMAKE_CURRENT_SOURCE_DIR}/cutlass/tools/util/include")
 include_directories("${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../")
 
 execute_process(COMMAND ${CMAKE_COMMAND} -E make_directory

diff --git a/paddle/phi/kernels/fusion/cutlass/gemm_epilogue/compile.sh b/paddle/phi/kernels/fusion/cutlass/gemm_epilogue/compile.sh
@@ -13,10 +13,10 @@
 # limitations under the License.
 set -e
 
-# cutlass_repo_directory="cutlass"
-# if [ ! -d "$cutlass_repo_directory" ]; then
-#     git clone --branch v2.11.0  https://github.com/NVIDIA/cutlass
-# fi
+cutlass_repo_directory="cutlass"
+if [ ! -d "$cutlass_repo_directory" ]; then
+    git clone --branch v2.11.0  https://github.com/NVIDIA/cutlass
+fi
 
 build_directory="build"
 if [ ! -d "$build_directory" ]; then