From f47472603b643b7b2b854699056d67b9ec603e49 Mon Sep 17 00:00:00 2001
From: Antonio Sanchez <cantonios@google.com>
Date: Mon, 9 Jan 2023 11:21:13 -0800
Subject: [PATCH] Add missing header for GPU tests.

---
 test/gpu_test_helper.h | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/test/gpu_test_helper.h b/test/gpu_test_helper.h
index 8955cbc5c..008e95cbe 100644
--- a/test/gpu_test_helper.h
+++ b/test/gpu_test_helper.h
@@ -3,6 +3,9 @@
 
 #include <Eigen/Core>
 
+// Allow gpu** macros for generic tests.
+#include <unsupported/Eigen/CXX11/src/Tensor/TensorGpuHipCudaDefines.h>
+
 // std::tuple cannot be used on device, and there is a bug in cuda < 9.2 that
 // doesn't allow std::tuple to compile for host code either. In these cases,
 // use our custom implementation.