diff --git a/tests/cpp/common/test_span.h b/tests/cpp/common/test_span.h
index 11a67caab..a53d4300d 100644
--- a/tests/cpp/common/test_span.h
+++ b/tests/cpp/common/test_span.h
@@ -99,7 +99,7 @@ struct TestRBeginREnd {
 
     Span<float> s (arr);
 
-#if defined(__CUDA_ARCH__)
+#if defined(__CUDA_ARCH__) || defined(__HIP_PLATFORM_AMD__)
     auto rbeg = dh::trbegin(s);
     auto rend = dh::trend(s);
 #else