| @@ -798,34 +798,6 @@ endif() | |||||
| if(MGE_WITH_CAMBRICON) | if(MGE_WITH_CAMBRICON) | ||||
| include_directories("$ENV{NEUWARE_HOME}/include") | include_directories("$ENV{NEUWARE_HOME}/include") | ||||
| link_directories("$ENV{NEUWARE_HOME}/lib64") | 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/cnrt.cmake) | ||||
| include(cmake/cndev.cmake) | include(cmake/cndev.cmake) | ||||
| include(cmake/cnml.cmake) | include(cmake/cnml.cmake) | ||||
| @@ -957,7 +929,6 @@ set(MEGDNN_WITH_ROCM ${MGE_WITH_ROCM}) | |||||
| # CAMBRICON | # CAMBRICON | ||||
| set(MGB_CAMBRICON ${MGE_WITH_CAMBRICON}) | set(MGB_CAMBRICON ${MGE_WITH_CAMBRICON}) | ||||
| set(MEGDNN_WITH_CAMBRICON ${MGE_WITH_CAMBRICON}) | |||||
| # ENFLAME | # ENFLAME | ||||
| set(MGB_ENFLAME ${MGE_WITH_ENFLAME}) | set(MGB_ENFLAME ${MGE_WITH_ENFLAME}) | ||||
| @@ -163,19 +163,6 @@ if(MGE_WITH_CUDA) | |||||
| list(APPEND SOURCES ${CUSOURCES}) | list(APPEND SOURCES ${CUSOURCES}) | ||||
| endif() | endif() | ||||
| if(MGE_WITH_CAMBRICON) | |||||
| file(GLOB_RECURSE SOURCES_ cambricon/*.cpp) | |||||
| list(APPEND SOURCES ${SOURCES_}) | |||||
| file(GLOB_RECURSE BANG_SOURCES cambricon/*.mlu) | |||||
| list(APPEND MEGDNN_INCLUDES "${PROJECT_SOURCE_DIR}/dnn/include") | |||||
| list(APPEND MEGDNN_INCLUDES "${PROJECT_SOURCE_DIR}/dnn") | |||||
| list(APPEND MEGDNN_INCLUDES "${PROJECT_BINARY_DIR}/genfiles") | |||||
| bang_compile(BANG_OBJS "${BANG_SOURCES}" "${MEGDNN_INCLUDES}") | |||||
| list(APPEND SOURCES ${BANG_OBJS}) | |||||
| endif() | |||||
| if(MGE_WITH_ATLAS) | if(MGE_WITH_ATLAS) | ||||
| file(GLOB_RECURSE SOURCES_ atlas/*.cpp) | file(GLOB_RECURSE SOURCES_ atlas/*.cpp) | ||||
| list(APPEND SOURCES ${SOURCES_}) | list(APPEND SOURCES ${SOURCES_}) | ||||
| @@ -264,8 +251,4 @@ if(CMAKE_THREAD_LIBS_INIT) | |||||
| target_link_libraries(megdnn PRIVATE Threads::Threads) | target_link_libraries(megdnn PRIVATE Threads::Threads) | ||||
| endif() | endif() | ||||
| if(MGE_WITH_CAMBRICON) | |||||
| target_link_libraries(megdnn PRIVATE ${BANG_OBJS} ${MGE_CAMBRICON_LIBS}) | |||||
| endif() | |||||
| install(TARGETS megdnn EXPORT ${MGE_EXPORT_TARGETS}) | install(TARGETS megdnn EXPORT ${MGE_EXPORT_TARGETS}) | ||||
| @@ -29,11 +29,6 @@ if(MGE_WITH_MIDOUT_PROFILE) | |||||
| list(APPEND SOURCES ${PROJECT_SOURCE_DIR}/third_party/midout/src/midout.cpp) | list(APPEND SOURCES ${PROJECT_SOURCE_DIR}/third_party/midout/src/midout.cpp) | ||||
| endif() | endif() | ||||
| if(MGE_WITH_CAMBRICON) | |||||
| file(GLOB_RECURSE SOURCES_ cambricon/*.cpp) | |||||
| list(APPEND SOURCES ${SOURCES_}) | |||||
| endif() | |||||
| if(MGE_WITH_ATLAS) | if(MGE_WITH_ATLAS) | ||||
| file(GLOB_RECURSE SOURCES_ atlas/*.cpp) | file(GLOB_RECURSE SOURCES_ atlas/*.cpp) | ||||
| list(APPEND SOURCES ${SOURCES_}) | list(APPEND SOURCES ${SOURCES_}) | ||||
| @@ -26,10 +26,6 @@ | |||||
| #include "megcore_rocm.h" | #include "megcore_rocm.h" | ||||
| #endif | #endif | ||||
| #if MGB_CAMBRICON | |||||
| #include "megcore_cambricon.h" | |||||
| #endif | |||||
| #if MGB_ATLAS | #if MGB_ATLAS | ||||
| #include "acl/acl.h" | #include "acl/acl.h" | ||||
| #include "megcore_atlas.h" | #include "megcore_atlas.h" | ||||
| @@ -81,10 +77,6 @@ MegDNNHandle::MegDNNHandle(const CompNodeEnv& env) { | |||||
| #if MGB_CAMBRICON | #if MGB_CAMBRICON | ||||
| if (env.property().type == CompNode::DeviceType::CAMBRICON) { | if (env.property().type == CompNode::DeviceType::CAMBRICON) { | ||||
| CompNodeEnv::CnrtEnv::init_status.init(); | CompNodeEnv::CnrtEnv::init_status.init(); | ||||
| megcore::createDeviceHandleWithGlobalInitStatus( | |||||
| &m_dev_hdl, env.cnrt_env().device, 0, true); | |||||
| megcore::createComputingHandleWithCambriconContext( | |||||
| &m_comp_hdl, m_dev_hdl, 0, {env.cnrt_env().queue}); | |||||
| init = true; | init = true; | ||||
| } | } | ||||
| #endif | #endif | ||||