From 7bb8ae138bcd7720a5b997cb8ba6a70ac6fcde22 Mon Sep 17 00:00:00 2001 From: Mark Kim Date: Thu, 10 Oct 2024 12:02:32 -0500 Subject: [PATCH] hip_fp8.h doesn't exist before hip 6.2 Should check before including. --- library/include/hipblaslt.h | 2 +- .../Source/lib/include/Tensile/DataTypes_Float8_BFloat8.hpp | 2 ++ tensilelite/Tensile/Source/lib/include/Tensile/hip_f8_impl.h | 2 ++ 3 files changed, 5 insertions(+), 1 deletion(-) diff --git a/library/include/hipblaslt.h b/library/include/hipblaslt.h index 41ff517f22..aa636ce486 100644 --- a/library/include/hipblaslt.h +++ b/library/include/hipblaslt.h @@ -69,7 +69,7 @@ #undef ROCM_USE_FLOAT8 #endif -#if defined(__HIPCC__) +#if defined(__HIPCC__) && defined(HIP_FP8_TYPE_OCP) #include #endif diff --git a/tensilelite/Tensile/Source/lib/include/Tensile/DataTypes_Float8_BFloat8.hpp b/tensilelite/Tensile/Source/lib/include/Tensile/DataTypes_Float8_BFloat8.hpp index b1a3320a28..c248b6d164 100644 --- a/tensilelite/Tensile/Source/lib/include/Tensile/DataTypes_Float8_BFloat8.hpp +++ b/tensilelite/Tensile/Source/lib/include/Tensile/DataTypes_Float8_BFloat8.hpp @@ -37,7 +37,9 @@ #define HIP_HOST __host__ #define HIP_DEVICE __device__ +#if defined(HIP_FP8_TYPE_OCP) #include +#endif namespace tensile_hip_f8_impl { diff --git a/tensilelite/Tensile/Source/lib/include/Tensile/hip_f8_impl.h b/tensilelite/Tensile/Source/lib/include/Tensile/hip_f8_impl.h index c5c0ea9b1b..f45720e06e 100644 --- a/tensilelite/Tensile/Source/lib/include/Tensile/hip_f8_impl.h +++ b/tensilelite/Tensile/Source/lib/include/Tensile/hip_f8_impl.h @@ -32,7 +32,9 @@ && HIP_VERSION_PATCH > 42130 //tmp before gfx94 use hip f8 header #define HIP_FP8_TYPE_OCP 1 #endif +#if defined(HIP_FP8_TYPE_OCP) #include +#endif namespace tensile_hip_f8_impl {