From 5c752a02b7788a69e91296f42944af2ec9fd602d Mon Sep 17 00:00:00 2001
From: "PoYen, Chen" <PoYen.Chen@amd.com>
Date: Tue, 11 Jun 2024 14:55:45 +0000
Subject: [PATCH] Fix wrong pipeline args for fp8

---
 example/ck_tile/01_fmha/generate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/example/ck_tile/01_fmha/generate.py b/example/ck_tile/01_fmha/generate.py
index 32ac200a26..6d667ff1d8 100644
--- a/example/ck_tile/01_fmha/generate.py
+++ b/example/ck_tile/01_fmha/generate.py
@@ -1001,7 +1001,7 @@ def get_fwd_splitkv_combine_blobs(kernel_filter : Optional[str], receipt) -> Lis
                 pipelines.append(Pipeline('unused', spad, dvpad, lse, squant))
         elif dtype in ['fp8', 'bf8']:
             # no need lse kernels
-            pipelines.append(Pipeline('unused', 'f', 'f', False, squant))
+            pipelines.append(Pipeline('unused', 'f', 'f', 'f', squant))
         else:
             assert False
         return pipelines