diff --git a/src/gemm_v2.cpp b/src/gemm_v2.cpp index 791bd33071..b317ba170b 100644 --- a/src/gemm_v2.cpp +++ b/src/gemm_v2.cpp @@ -50,8 +50,12 @@ #else #include /// rocblas_gemm_ex3 supports F8 datatypes. +#ifdef _WIN32 +#define USE_ROCBLAS_GEMM_EX3 ((MIOPEN_ROCBLAS_VERSION_FLAT >= 3000000) && ROCBLAS_BETA_FEATURES_API) +#else #define USE_ROCBLAS_GEMM_EX3 ((MIOPEN_ROCBLAS_VERSION_FLAT >= 2047000) && ROCBLAS_BETA_FEATURES_API) #endif +#endif #include #endif