Fix for running rocminfo on wider variety of CPUs

Change-Id: I7acc46d2fec97ae1026643f79e59dd632444ce2e [ROCm/rocminfo commit: 9b0cd9e428]
2019-12-09 14:59:57 -06:00
parent 3f0aab31d5
commit 8eeb8d3e5f
1 changed files with 4 additions and 1 deletions
@@ -181,7 +181,10 @@ set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall")
 # Extend the compiler flags for 64-bit builds
 #
 if (IS64BIT)
-  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -m64  -msse -msse2")
+  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -m64")
+  if((${CMAKE_HOST_SYSTEM_PROCESSOR} STREQUAL "x86_64") OR (${CMAKE_HOST_SYSTEM_PROCESSOR} STREQUAL "AMD64"))
+    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -msse -msse2")
+  endif()
 else()
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -m32")
 endif()