| |
| IF(CMAKE_SYSTEM_NAME MATCHES "Linux") |
| message("-- <FindVSX>") |
| EXEC_PROGRAM(LD_SHOW_AUXV=1 ARGS "/bin/true" OUTPUT_VARIABLE bintrue) |
| if(bintrue MATCHES "AT_PLATFORM:[ \\t\\n\\r]*([a-zA-Z0-9_]+)[ \\t\\n\\r]*") |
| if(CMAKE_MATCH_COUNT GREATER 0) |
| string(TOLOWER ${CMAKE_MATCH_1} platform) |
| if(${platform} MATCHES "^power") |
| message("-- POWER Platform: ${platform}") |
| SET(POWER_COMP TRUE CACHE BOOL "power ") |
| SET(CXX_VSX_FLAGS "${CXX_VSX_FLAGS} -mcpu=${platform} -mtune=${platform}" ) |
| endif() |
| endif() |
| endif() |
| SET(VSX_CODE " #include <altivec.h> |
| int main() { |
| float __attribute__((aligned(16))) vptr_y[8] = { 1.0f,2.f,3.f,4.f,4.f,3.f,2.f,1.f }; |
| __vector float v_result = vec_add(vec_vsx_ld(0, vptr_y), vec_vsx_ld(16, vptr_y)); |
| return 0; |
| }") |
| #check_cxx_compiler_flag(-mvsx vsx_flag) |
| SET(CMAKE_REQUIRED_FLAGS_SAVE ${CMAKE_REQUIRED_FLAGS}) |
| SET(CMAKE_REQUIRED_FLAGS "-mvsx") |
| CHECK_C_SOURCE_COMPILES("${VSX_CODE}" C_VSX_FOUND) |
| CHECK_CXX_SOURCE_COMPILES("${VSX_CODE}" CXX_VSX_FOUND) |
| SET(CMAKE_REQUIRED_FLAGS ${CMAKE_REQUIRED_FLAGS_SAVE}) |
| if(CXX_VSX_FOUND) |
| message("-- VSX flag was set.") |
| SET(CXX_VSX_FLAGS "${CXX_VSX_FLAGS} -mvsx" ) |
| elseif(POWER_COMP) |
| message(WARNING "-- VSX flag was not set.") |
| endif() |
| message("-- </FindVSX>") |
| endif() |
