From 4aa640ca6d571d148e8adebd62625eeb451a7fe2 Mon Sep 17 00:00:00 2001 From: Bangtian Liu Date: Mon, 2 Dec 2024 01:12:12 -0600 Subject: [PATCH] [tuner]: fix the ci error Signed-off-by: Bangtian Liu --- tuner/tuner/common.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tuner/tuner/common.py b/tuner/tuner/common.py index b33d5845e..3253dd077 100644 --- a/tuner/tuner/common.py +++ b/tuner/tuner/common.py @@ -121,6 +121,7 @@ class Configuration: def intrinsic(self) -> Optional[iree_gpu.MMAAttr]: if "mma_kind" in self.lowering_config.attributes: return self.lowering_config.attributes["mma_kind"] + return None def tilesize_workgroup(self) -> list[int]: if "workgroup" in self.lowering_config.attributes: @@ -151,7 +152,7 @@ def get_lowering_config( tuner_ctx: TunerContext, **kwargs: Any, ) -> iree_gpu.LoweringConfigAttr: - lowering_config_dict = {} + lowering_config_dict: dict[str, Any] = {} for key, value in kwargs.items(): match key: case "workgroup" | "reduction":