Add GPU build guard around gpu_prim_helpers_test
diff --git a/tensorflow/core/kernels/gpu_prim_helpers_test.cu.cc b/tensorflow/core/kernels/gpu_prim_helpers_test.cu.cc
index fdb38d5..a3ad385 100644
--- a/tensorflow/core/kernels/gpu_prim_helpers_test.cu.cc
+++ b/tensorflow/core/kernels/gpu_prim_helpers_test.cu.cc
@@ -13,6 +13,8 @@
limitations under the License.
==============================================================================*/
+#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
+
#include "tensorflow/core/framework/fake_input.h"
#include "tensorflow/core/framework/node_def_builder.h"
#include "tensorflow/core/framework/op_kernel.h"
@@ -221,3 +223,5 @@
} // namespace
} // namespace tensorflow
+
+#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
