commit 372124e6af36a540e74a2ec31d79d7297a831f98
author Frédéric Bastien <frederic.bastien@gmail.com> Thu Aug 01 12:38:52 2024 -0700
committer Copybara-Service <copybara-worker@google.com> Thu Aug 01 12:40:25 2024 -0700
tree 85f4f2a49cefa0a1d16dd8cb14c9a8ca1f5457eb
parent d0159eab5985aa8561ccda9ee9987d6b2726fd76

PR #1732: Fix build on NVIDIA Jetson board. Fix #1665

Imported from GitHub PR https://github.com/abseil/abseil-cpp/pull/1732

Fix build on NVIDIA Jetson board. Fix #1665

This patch is already used by the spark project.
I'm fixing this as this break the build of Tensorflow and JAX on Jetson board.
Merge 7db2d2ab9fbed1f0fabad10a6ec73533ba71bfff into 6b8ebb35c0414ef5a2b6fd4a0f59057e41beaff9

Merging this change closes #1732

COPYBARA_INTEGRATE_REVIEW=https://github.com/abseil/abseil-cpp/pull/1732 from nouiz:fix_neon_on_jetson 7db2d2ab9fbed1f0fabad10a6ec73533ba71bfff
PiperOrigin-RevId: 658501520
Change-Id: If502ede4efc8c877fb3fed227eca6dc7622dd181

absl/base/config.h

diff --git a/absl/base/config.h b/absl/base/config.h
index 97c9a22..ab1e986 100644
--- a/absl/base/config.h
+++ b/absl/base/config.h
@@ -926,7 +926,7 @@
 // https://llvm.org/docs/CompileCudaWithLLVM.html#detecting-clang-vs-nvcc-from-code
 #ifdef ABSL_INTERNAL_HAVE_ARM_NEON
 #error ABSL_INTERNAL_HAVE_ARM_NEON cannot be directly set
-#elif defined(__ARM_NEON) && !defined(__CUDA_ARCH__)
+#elif defined(__ARM_NEON) && !(defined(__NVCC__) && defined(__CUDACC__))
 #define ABSL_INTERNAL_HAVE_ARM_NEON 1
 #endif