|
|
@@ -798,34 +798,6 @@ endif() |
|
|
|
if(MGE_WITH_CAMBRICON) |
|
|
|
include_directories("$ENV{NEUWARE_HOME}/include") |
|
|
|
link_directories("$ENV{NEUWARE_HOME}/lib64") |
|
|
|
include(cmake/FindBANG/FindBANG.cmake) |
|
|
|
if (${MGE_MLU_ARCH} STREQUAL "MLU100") |
|
|
|
set(BANG_ARCH "100") |
|
|
|
elseif (${MGE_MLU_ARCH} STREQUAL "MLU1h8") |
|
|
|
set(BANG_ARCH "110") |
|
|
|
elseif (${MGE_MLU_ARCH} STREQUAL "MLU220") |
|
|
|
set(BANG_ARCH "220") |
|
|
|
elseif (${MGE_MLU_ARCH} STREQUAL "MLU270") |
|
|
|
set(BANG_ARCH "270") |
|
|
|
elseif (${MGE_MLU_ARCH} STREQUAL "MLU290") |
|
|
|
set(BANG_ARCH "290") |
|
|
|
elseif (${MGE_MLU_ARCH} STREQUAL "MLU200") |
|
|
|
set(BANG_ARCH "200") |
|
|
|
else() |
|
|
|
message (FATAL_ERROR "Unsupported MLU arch.") |
|
|
|
endif() |
|
|
|
set(BANG_CNCC_FLAGS "${BANG_CNCC_FLAGS} --bang-mlu-arch=${MGE_MLU_ARCH}") |
|
|
|
set(BANG_CNCC_FLAGS "${BANG_CNCC_FLAGS} -std=c++11 -Werror") |
|
|
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -D__BANG_ARCH__=${BANG_ARCH}") |
|
|
|
if (${CMAKE_BUILD_TYPE} STREQUAL "Debug") |
|
|
|
set(BANG_CNCC_FLAGS "${BANG_CNCC_FLAGS} -g -O0") |
|
|
|
elseif (${CMAKE_BUILD_TYPE} STREQUAL "Release") |
|
|
|
set(BANG_CNCC_FLAGS "${BANG_CNCC_FLAGS} -O3") |
|
|
|
elseif (${CMAKE_BUILD_TYPE} STREQUAL "RelWithDebInfo") |
|
|
|
set(BANG_CNCC_FLAGS "${BANG_CNCC_FLAGS} -g -O3") |
|
|
|
elseif (${CMAKE_BUILD_TYPE} STREQUAL "MinSizeRel") |
|
|
|
set(BANG_CNCC_FLAGS "${BANG_CNCC_FLAGS} -Os") |
|
|
|
endif() |
|
|
|
include(cmake/cnrt.cmake) |
|
|
|
include(cmake/cndev.cmake) |
|
|
|
include(cmake/cnml.cmake) |
|
|
@@ -957,7 +929,6 @@ set(MEGDNN_WITH_ROCM ${MGE_WITH_ROCM}) |
|
|
|
|
|
|
|
# CAMBRICON |
|
|
|
set(MGB_CAMBRICON ${MGE_WITH_CAMBRICON}) |
|
|
|
set(MEGDNN_WITH_CAMBRICON ${MGE_WITH_CAMBRICON}) |
|
|
|
|
|
|
|
# ENFLAME |
|
|
|
set(MGB_ENFLAME ${MGE_WITH_ENFLAME}) |
|
|
|