Skip to content

Commit

Permalink
arch list fix in intel_cpu cmakelists
Browse files Browse the repository at this point in the history
  • Loading branch information
NishantPrabhuFujitsu committed Dec 12, 2024
1 parent d17c350 commit a81bf1e
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions src/plugins/intel_cpu/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -280,8 +280,8 @@ target_include_directories(${TARGET_NAME} PRIVATE $<TARGET_PROPERTY:openvino::re

# ARCH lists for softmax.cpp and mha_single_token.cpp
# Based on result of above calls, decide whether to add SVE
set(SOFTMAX_ARCH_LIST AVX512F AVX2 ANY)
set(MHA_SINGLE_TOKEN_ARCH_LIST AVX512F AVX2 ANY)
set(SOFTMAX_ARCH_LIST AVX512F AVX2)
set(MHA_SINGLE_TOKEN_ARCH_LIST AVX512F AVX2)

if(ENABLE_NEON_FP16)
list(APPEND SOFTMAX_ARCH_LIST NEON_FP16)
Expand All @@ -293,6 +293,9 @@ if(ENABLE_SVE)
list(APPEND MHA_SINGLE_TOKEN_ARCH_LIST SVE)
endif()

list(APPEND SOFTMAX_ARCH_LIST ANY)
list(APPEND MHA_SINGLE_TOKEN_ARCH_LIST ANY)

# Cross compiled function
# TODO: The same for proposal, proposalONNX, topk
cross_compiled_file(${TARGET_NAME}
Expand Down

0 comments on commit a81bf1e

Please sign in to comment.