From b4dcd5803f1dae92467d39c31f176131ce796735 Mon Sep 17 00:00:00 2001
From: Nicolas Macchioni <nmacchioni@fb.com>
Date: Fri, 8 Dec 2023 11:30:01 -0800
Subject: [PATCH] Add F8 dtype definition in f16_f8_f16 gemm instances (#1092)

---
 .../device_gemm_xdl_c_shuffle_f16_f8_f16_mk_kn_mn_instance.cpp   | 1 +
 .../device_gemm_xdl_c_shuffle_f16_f8_f16_mk_nk_mn_instance.cpp   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/library/src/tensor_operation_instance/gpu/gemm/device_gemm_xdl_c_shuffle_f16_f8_f16_mk_kn_mn_instance.cpp b/library/src/tensor_operation_instance/gpu/gemm/device_gemm_xdl_c_shuffle_f16_f8_f16_mk_kn_mn_instance.cpp
index 3c9e03b674..38667ad42b 100644
--- a/library/src/tensor_operation_instance/gpu/gemm/device_gemm_xdl_c_shuffle_f16_f8_f16_mk_kn_mn_instance.cpp
+++ b/library/src/tensor_operation_instance/gpu/gemm/device_gemm_xdl_c_shuffle_f16_f8_f16_mk_kn_mn_instance.cpp
@@ -16,6 +16,7 @@ namespace tensor_operation {
 namespace device {
 namespace instance {
 
+using F8 = ck::f8_t;
 using F16 = ck::half_t;
 using F32 = float;
 
diff --git a/library/src/tensor_operation_instance/gpu/gemm/device_gemm_xdl_c_shuffle_f16_f8_f16_mk_nk_mn_instance.cpp b/library/src/tensor_operation_instance/gpu/gemm/device_gemm_xdl_c_shuffle_f16_f8_f16_mk_nk_mn_instance.cpp
index aab0af990d..820404e064 100644
--- a/library/src/tensor_operation_instance/gpu/gemm/device_gemm_xdl_c_shuffle_f16_f8_f16_mk_nk_mn_instance.cpp
+++ b/library/src/tensor_operation_instance/gpu/gemm/device_gemm_xdl_c_shuffle_f16_f8_f16_mk_nk_mn_instance.cpp
@@ -16,6 +16,7 @@ namespace tensor_operation {
 namespace device {
 namespace instance {
 
+using F8 = ck::f8_t;
 using F16 = ck::half_t;
 using F32 = float;