xref: /aosp_15_r20/external/pytorch/cmake/Modules/FindVSX.cmake (revision da0073e96a02ea20f0ac840b70461e3646d07c45)

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()