Linux Build: Split kernel lib into smaller libs (#1066)

Split kernel lib into smaller libs.
intel · Nov 11, 2024 · e250a26 · e250a26
2 parents 92bdb91 + f713077
commit e250a26
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/BuildOnLinux.cmake b/src/BuildOnLinux.cmake
@@ -34,7 +34,10 @@ else()
   foreach(sycl_src ${ATen_XPU_SYCL_SRCS})
     string(REGEX MATCH "Binary" IS_BINARY ${sycl_src})
     string(REGEX MATCH "Unary" IS_UNARY ${sycl_src})
+    string(REGEX MATCH "Pow" IS_POW ${sycl_src})
+    string(REGEX MATCH "Copy" IS_COPY ${sycl_src})
     string(REGEX MATCH "Reduce" IS_REDUCE ${sycl_src})
+    string(REGEX MATCH "Activation" IS_ACTIVATION ${sycl_src})
     string(REGEX MATCH "Foreach" IS_FOREACH ${sycl_src})
 
     if(NOT IS_FOREACH STREQUAL "")
@@ -43,6 +46,10 @@ else()
       list(APPEND ATen_XPU_SYCL_REDUCE_SRCS ${sycl_src})
     elseif(NOT IS_UNARY STREQUAL "" OR NOT IS_BINARY STREQUAL "")
       list(APPEND ATen_XPU_SYCL_UNARY_BINARY_SRCS ${sycl_src})
+    elseif(NOT IS_COPY STREQUAL "" OR NOT IS_POW STREQUAL "")
+      list(APPEND ATen_XPU_SYCL_UNARY_BINARY_SRCS ${sycl_src})
+    elseif(NOT IS_ACTIVATION STREQUAL "")
+      list(APPEND ATen_XPU_SYCL_UNARY_BINARY_SRCS ${sycl_src})
     else()
       list(APPEND ATen_XPU_SYCL_OTHERS_SRCS ${sycl_src})
     endif()