From a1e26e4298aac041e56c856c40bcb0c29fbc3f83 Mon Sep 17 00:00:00 2001
From: Gunhan Gulsoy <gunan@google.com>
Date: Wed, 3 Jun 2020 16:05:26 -0700
Subject: [PATCH] Avoid using cuda specific GPU function to unbreak ROCM build.

Indirect them through core/util/gpu_functions.h

PiperOrigin-RevId: 314622940
Change-Id: I0e5a349f759d0af6ff13acc43b34080a5104c9cc
---
 tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc b/tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc
index 625e03b4352..7fb2d3916e3 100644
--- a/tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc
+++ b/tensorflow/core/kernels/scatter_nd_op_gpu.cu.cc
@@ -58,14 +58,14 @@ struct LeftUpdate<T, scatter_nd_op::UpdateOp::SUB> {
 template <typename T>
 struct LeftUpdate<T, scatter_nd_op::UpdateOp::MAX> {
   EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC void operator()(T* out, const T& val) {
-    CudaAtomicMax(out, val);
+    GpuAtomicMax(out, val);
   }
 };
 
 template <typename T>
 struct LeftUpdate<T, scatter_nd_op::UpdateOp::MIN> {
   EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC void operator()(T* out, const T& val) {
-    CudaAtomicMin(out, val);
+    GpuAtomicMin(out, val);
   }
 };