make examples/models/llama/runner build in win_x64

Author: Yi Ren

examples/models/llama/CMakeLists.txt

@@ -98,14 +98,21 @@ add_subdirectory(runner)
 set(link_libraries gflags)
 set(_srcs main.cpp)
 
+# in Windows, cpublas and extension_threadpool have duplicated symbols
+if(WIN32 AND TARGET xnnpack_backend)
+  set(CPUBLAS "")
+else()
+  set(CPUBLAS "cpublas")
+endif()
+
 if(EXECUTORCH_BUILD_KERNELS_OPTIMIZED)
   list(
     APPEND
     link_libraries
     optimized_native_cpu_ops_lib
     optimized_kernels
     portable_kernels
-    cpublas
+    ${CPUBLAS}
     eigen_blas
   )
   target_link_options_shared_lib(optimized_native_cpu_ops_lib)
@@ -131,18 +138,26 @@ if(EXECUTORCH_BUILD_TORCHAO)
 endif()
 
 set(XNNPACK_ROOT ${CMAKE_CURRENT_SOURCE_DIR}/../../../backends/xnnpack)
+
+# in Windows, xnnpack_backend and extension_threadpool have duplicated symbols
+if(WIN32 AND TARGET xnnpack_backend)
+  set(EXTENSION_THREADPOOL "")
+else()
+  set(EXTENSION_THREADPOOL "extension_threadpool")
+endif()
+
 # Extra compile option and include dir for pthreadpool
 if(EXECUTORCH_BUILD_PTHREADPOOL)
   list(APPEND _common_compile_options -DET_USE_THREADPOOL)
-  list(APPEND link_libraries extension_threadpool pthreadpool)
+  list(APPEND link_libraries ${EXTENSION_THREADPOOL} pthreadpool)
   list(APPEND _common_include_directories
        ${XNNPACK_ROOT}/third-party/pthreadpool/include
   )
 endif()
 
 # Extra sources for cpuinfo
 if(EXECUTORCH_BUILD_CPUINFO)
-  list(APPEND link_libraries extension_threadpool cpuinfo)
+  list(APPEND link_libraries ${EXTENSION_THREADPOOL} cpuinfo)
   list(APPEND _common_include_directories
        ${XNNPACK_ROOT}/third-party/cpuinfo/include
   )

examples/models/llama/runner/CMakeLists.txt

@@ -52,6 +52,7 @@ list(APPEND _llama_runner__srcs
 if(CMAKE_TOOLCHAIN_IOS
    OR ANDROID
    OR APPLE
+  OR WIN32
 )
   # Building a share library on iOS requires code signing On Android we see
   # duplicated registration when using shared lib
@@ -75,13 +76,24 @@ add_subdirectory(
 )
 set(CMAKE_POSITION_INDEPENDENT_CODE ${_pic_flag})
 
+if (WIN32)
+  # in Windows, only extension_module_static works and has duplicated symbols with extension_tensor
+  set(llama_runner_deps executorch extension_data_loader extension_module_static
+                        re2::re2
+  )
+  # in Windows, need PUBLIC or sources files in executorch/examples/models/llama/runner/ cannot be found
+  target_include_directories(
+    llama_runner PUBLIC ${_common_include_directories} ${EXECUTORCH_ROOT}
+  )
+else()
 set(llama_runner_deps executorch extension_data_loader extension_module
                       extension_tensor re2::re2
 )
-
-target_link_libraries(llama_runner PUBLIC ${llama_runner_deps})
-
 target_include_directories(
   llama_runner INTERFACE ${_common_include_directories} ${EXECUTORCH_ROOT}
 )
+endif()
+
+target_link_libraries(llama_runner PUBLIC ${llama_runner_deps})
+
 target_compile_options(llama_runner PUBLIC ${_preprocessor_flag})