From 8eb6e451f2a65bd506f194982811ddc1471bc6f2 Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Tue, 11 Jun 2024 14:37:18 +0000 Subject: [PATCH] Undo disabling data types --- example/ck_tile/01_fmha/generate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 = {