Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[tuner]: use lowering config binding #629

Merged
merged 9 commits into from
Dec 3, 2024
Prev Previous commit
Next Next commit
[tuner]: fix the ci error
Signed-off-by: Bangtian Liu <[email protected]>
bangtianliu committed Dec 3, 2024
commit a2094ac8187bfde94869cb311a0bbf31d8168410
3 changes: 2 additions & 1 deletion tuner/tuner/common.py
Original file line number Diff line number Diff line change
@@ -121,6 +121,7 @@ class Configuration:
def intrinsic(self) -> Optional[iree_gpu.MMAAttr]:
bangtianliu marked this conversation as resolved.
Show resolved Hide resolved
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":
bangtianliu marked this conversation as resolved.
Show resolved Hide resolved