From 4aaa257a5dad64215e31ae57b5a394682a40bc35 Mon Sep 17 00:00:00 2001
From: Jeffrey Poznanovic <Jeffrey.Poznanovic@amd.com>
Date: Tue, 19 Mar 2019 04:13:03 +0000
Subject: [PATCH] Added ROCm support for nn_ops_test

---
 tensorflow/core/kernels/nn_ops_test.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tensorflow/core/kernels/nn_ops_test.cc b/tensorflow/core/kernels/nn_ops_test.cc
index a841291ddd7..af4f7d58f6a 100644
--- a/tensorflow/core/kernels/nn_ops_test.cc
+++ b/tensorflow/core/kernels/nn_ops_test.cc
@@ -15,9 +15,9 @@ limitations under the License.
 
 #define EIGEN_USE_THREADS
 
-#if GOOGLE_CUDA
+#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
 #define EIGEN_USE_GPU
-#endif  // GOOGLE_CUDA
+#endif  // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
 
 #include <functional>
 #include <memory>