Update tensorflow-xla-runtime.spec

cms-sw · Mar 15, 2024 · 398bf38 · 398bf38
1 parent 86827d9
commit 398bf38
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tensorflow-xla-runtime.spec b/tensorflow-xla-runtime.spec
@@ -19,6 +19,9 @@ CXXFLAGS="-fPIC %{arch_build_flags} ${CMS_EIGEN_CXX_FLAGS}"
 %ifarch x86_64
     CXXFLAGS="${CXXFLAGS} -msse3"
 %endif
+%ifarch aarch64
+    CXXFLAGS="${CXXFLAGS} -DEIGEN_NEON_GEBP_NR=4"
+%endif
 
 pushd tensorflow/xla_aot_runtime_src
   cmake . -DCMAKE_CXX_FLAGS="${CXXFLAGS}" -DCMAKE_CXX_STANDARD=%{cms_cxx_standard} -DBUILD_SHARED_LIBS=OFF