Fix dpp gemm

dff73bcc · Rostyslav Geyyer · cde27d1f · dff73bcc
Commit dff73bcc authored Oct 06, 2023 by Rostyslav Geyyer
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

include/ck/tensor_operation/gpu/warp/dpp_gemm.hpp include/ck/tensor_operation/gpu/warp/dpp_gemm.hpp +5 -1

No files found.
--- a/include/ck/tensor_operation/gpu/warp/dpp_gemm.hpp
+++ b/include/ck/tensor_operation/gpu/warp/dpp_gemm.hpp
@@ -450,7 +450,11 @@ struct DppGemm
    {
        static_assert(is_same<BaseType, double>::value || is_same<BaseType, float>::value ||
                          is_same<BaseType, half_t>::value || is_same<BaseType, bhalf_t>::value ||
-                          is_same<BaseType, int8_t>::value || is_same<BaseType, f8_t>::value,
+                          is_same<BaseType, int8_t>::value
+#if defined CK_ENABLE_FP8
+                          || is_same<base_type, f8_t>::value
+#endif
+                      ,
                      "base BaseType must be double, float, half, bfloat16, and int8_t!");

        static_for<0, KPack / dpp_instr.k_per_dpp, 1>{}([&](auto k) {