modify codebase cmakelist

open-mmlab · Apr 26, 2023 · 3e732f6 · 3e732f6
1 parent 0f43de9
commit 3e732f6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/csrc/mmdeploy/codebase/CMakeLists.txt b/csrc/mmdeploy/codebase/CMakeLists.txt
@@ -18,5 +18,10 @@ endif ()
 
 foreach (codebase IN LISTS CODEBASES)
     message(STATUS "build codebase: ${codebase}")
-    add_subdirectory(${codebase})
+    if (codebase STREQUAL "mmpretrain")
+        set(subdir_name "mmcls")
+    else()
+	set(subdir_name ${codebase})
+    endif()
+    add_subdirectory(${subdir_name})
 endforeach ()