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":