From 3e4a587135447f4ba0f64010870acc297d333198 Mon Sep 17 00:00:00 2001
From: Ramya Ramineni <62723901+rraminen@users.noreply.github.com>
Date: Thu, 19 Oct 2023 13:14:01 -0500
Subject: [PATCH] Added rocblas header (#4538)

Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
---
 csrc/includes/gemm_test.h | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/csrc/includes/gemm_test.h b/csrc/includes/gemm_test.h
index cd9fbb5a4..6d04921c2 100644
--- a/csrc/includes/gemm_test.h
+++ b/csrc/includes/gemm_test.h
@@ -9,6 +9,9 @@
 #ifndef __HIP_PLATFORM_HCC__
 #include <cuda_profiler_api.h>
 #endif
+#ifdef __HIP_PLATFORM_HCC__
+#include <rocblas/rocblas.h>
+#endif
 #include <array>
 #include <cstdio>
 #include <cstdlib>