Update CMake config

Author: shadaj

Examples/LeNet-MNIST/CMakeLists.txt

@@ -2,6 +2,7 @@ add_executable(LeNet-MNIST
   main.swift)
 target_link_libraries(LeNet-MNIST PRIVATE
   ImageClassificationModels
+  LayerInit
   Datasets)
 
 

Models/CMakeLists.txt

@@ -1,3 +1,4 @@
 add_subdirectory(ImageClassification)
+add_subdirectory(LayerInit)
 add_subdirectory(Recommendation)
 add_subdirectory(Text)

Models/ImageClassification/CMakeLists.txt

@@ -15,7 +15,8 @@ set_target_properties(ImageClassificationModels PROPERTIES
   INTERFACE_INCLUDE_DIRECTORIES ${CMAKE_Swift_MODULE_DIRECTORY})
 target_compile_options(ImageClassificationModels PRIVATE
   $<$<BOOL:${BUILD_TESTING}>:-enable-testing>)
-
+target_link_libraries(ImageClassificationModels PUBLIC
+  LayerInit)
 
 install(TARGETS ImageClassificationModels
   ARCHIVE DESTINATION lib/swift/$<LOWER_CASE:${CMAKE_SYSTEM_NAME}>

Models/LayerInit/CMakeLists.txt

@@ -0,0 +1,22 @@
+add_library(LayerInit
+  AutoConv.swift
+  AutoDense.swift
+  AutoFlatten.swift
+  AutoLayer.swift
+  AutoPool.swift
+  AutoSequenced.swift)
+set_target_properties(LayerInit PROPERTIES
+  INTERFACE_INCLUDE_DIRECTORIES ${CMAKE_Swift_MODULE_DIRECTORY})
+target_compile_options(LayerInit PRIVATE
+  $<$<BOOL:${BUILD_TESTING}>:-enable-testing>)
+
+
+install(TARGETS LayerInit
+  ARCHIVE DESTINATION lib/swift/$<LOWER_CASE:${CMAKE_SYSTEM_NAME}>
+  LIBRARY DESTINATION lib/swift/$<LOWER_CASE:${CMAKE_SYSTEM_NAME}>
+  RUNTIME DESTINATION bin)
+get_swift_host_arch(swift_arch)
+install(FILES
+  $<TARGET_PROPERTY:LayerInit,Swift_MODULE_DIRECTORY>/LayerInit.swiftdoc
+  $<TARGET_PROPERTY:LayerInit,Swift_MODULE_DIRECTORY>/LayerInit.swiftmodule
+  DESTINATION lib/swift$<$<NOT:$<BOOL:${BUILD_SHARED_LIBS}>>:_static>/$<LOWER_CASE:${CMAKE_SYSTEM_NAME}>/${swift_arch})