diff --git a/example/ck_tile/01_fmha/generate.py b/example/ck_tile/01_fmha/generate.py
index 7dccc6bc98..1e0ab127d3 100644
--- a/example/ck_tile/01_fmha/generate.py
+++ b/example/ck_tile/01_fmha/generate.py
@@ -12,8 +12,8 @@ import fnmatch
 
 DTYPE_MAP = {
     "fp16": "ck_tile::fp16_t",
-    # "bf16": "ck_tile::bf16_t",
-    # "fp8" : "ck_tile::fp8_t"
+    "bf16": "ck_tile::bf16_t",
+    "fp8" : "ck_tile::fp8_t"
 }
 
 DTYPE_BITS = {