Browse Source

fix(mgb/build): do not install cutlass

GitOrigin-RevId: ba8b047659
tags/v1.0.0-rc1
Megvii Engine Team 4 years ago
parent
commit
59dcd3b7f3
2 changed files with 1 additions and 2 deletions
  1. +0
    -1
      dnn/CMakeLists.txt
  2. +1
    -1
      dnn/src/CMakeLists.txt

+ 0
- 1
dnn/CMakeLists.txt View File

@@ -62,7 +62,6 @@ if(MGE_WITH_CUDA)
target_include_directories(cutlass
INTERFACE
$<BUILD_INTERFACE:${PROJECT_SOURCE_DIR}/third_party/cutlass/include>)
install(TARGETS cutlass EXPORT ${MGE_EXPORT_TARGETS})
endif()

if(MGE_WITH_TEST)


+ 1
- 1
dnn/src/CMakeLists.txt View File

@@ -130,7 +130,7 @@ add_library(megdnn EXCLUDE_FROM_ALL OBJECT ${SOURCES})

target_link_libraries(megdnn PUBLIC opr_param_defs)
if(MGE_WITH_CUDA)
target_link_libraries(megdnn PUBLIC cutlass)
target_link_libraries(megdnn PRIVATE $<BUILD_INTERFACE:cutlass>)
endif()

if(MGE_WITH_ROCM)


Loading…
Cancel
Save