[ATLAS] Update CMakeLists.txt for Atlas Platform

Tencent · Feb 29, 2024 · 6be1b0d · 6be1b0d
1 parent 97e7c1b
commit 6be1b0d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -429,6 +429,7 @@ if(TNN_HUAWEI_NPU_ENABLE)
     set(TARGET_OBJECTS ${TARGET_OBJECTS} "$<TARGET_OBJECTS:TNNNPU>")
 endif()
 if(TNN_ATLAS_ENABLE)
+    add_definitions(-DGET_NETWORK_ENABLE)
     add_subdirectory(source/tnn/device/atlas)
     set(TARGET_OBJECTS ${TARGET_OBJECTS} "$<TARGET_OBJECTS:TNNAtlas>")
     include_directories(${CMAKE_SOURCE_DIR}/source/tnn/device/atlas)