commit df5f032ac464ed497245cfc3ed14ca83def947a5
author Ashutosh Sharma <ashutosh.sharma@os.amperecomputing.com> Fri Oct 22 11:23:35 2021 -0700
committer Ashutosh Sharma <ashutosh.sharma@os.amperecomputing.com> Fri Oct 22 11:23:35 2021 -0700
tree 15b15f14545e39dd4fd5993cad09186e8b4ea150
parent 42df63f59dcbc523f7c3f69f165f6e71413ea3db

Removed NUMA check for ARM64

tensorflow/stream_executor/cuda/cuda_gpu_executor.cc

diff --git a/tensorflow/stream_executor/cuda/cuda_gpu_executor.cc b/tensorflow/stream_executor/cuda/cuda_gpu_executor.cc
index 93550b9..c183c5e 100644
--- a/tensorflow/stream_executor/cuda/cuda_gpu_executor.cc
+++ b/tensorflow/stream_executor/cuda/cuda_gpu_executor.cc
@@ -902,9 +902,6 @@
 #elif defined(PLATFORM_WINDOWS)
   // Windows support for NUMA is not currently implemented. Return node 0.
   return 0;
-#elif defined(__aarch64__)
-  LOG(INFO) << "ARM64 does not support NUMA - returning NUMA node zero";
-  return 0;
 #else
   VLOG(2) << "trying to read NUMA node for device ordinal: " << device_ordinal;
   static const int kUnknownNumaNode = -1;