From f236a13d1b9c1d8a8e572ab441d62cdc8604a373 Mon Sep 17 00:00:00 2001 From: carlushuang Date: Thu, 28 Mar 2024 22:00:11 +0000 Subject: [PATCH] fix several issue --- example/ck_tile/01_fmha/generate.py | 2 +- example/ck_tile/01_fmha/utils.hpp | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/example/ck_tile/01_fmha/generate.py b/example/ck_tile/01_fmha/generate.py index 5c44ad303b..2c50f47bd7 100644 --- a/example/ck_tile/01_fmha/generate.py +++ b/example/ck_tile/01_fmha/generate.py @@ -428,7 +428,7 @@ def get_blobs(kernel_filter : Optional[str]) -> tuple[FmhaFwdApiPool, List[FmhaF for mask, bias in itertools.product(MASK_MAP.keys(), ["t", "f"]): pipelines.append(FmhaFwdPipeline('qr_fp8', 'col', 'f', 'f', 'f', 'f', bias, 'f', mask)) else: - assert Fasle + assert False return pipelines gen = list() diff --git a/example/ck_tile/01_fmha/utils.hpp b/example/ck_tile/01_fmha/utils.hpp index 14347a344c..e10ae617dc 100644 --- a/example/ck_tile/01_fmha/utils.hpp +++ b/example/ck_tile/01_fmha/utils.hpp @@ -9,6 +9,7 @@ #include #include #include +#include #include "ck_tile/core/container/span.hpp"