ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 14:29:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix type hint (#2254)

Browse Source
This commit is contained in:
Casey-Shi
2025-05-28 23:43:58 +08:00
committed by GitHub
parent 9bd01b624e
commit 4286eae09a
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tile_engine/ops/gemm/json_config.py
View File

@@ -9,7 +9,7 @@ Handles loading, parsing, and validation of JSON configuration parameters.
from pathlib import Path
from dataclasses import dataclass
from typing import List, Optional, Union, Tuple, Type
from typing import List, Optional, Union, Tuple, Type, Dict
import json
@@ -63,8 +63,8 @@ class ProblemConfig:
layouts: Tuple[EnumConfigParam, ...]
@property
def datatype_map(self) -> dict[str, str]:
"""Get current layout selections as a key-value map."""
def datatype_map(self) -> Dict[str, str]:
"""Get datatype as a key-value map."""
return {
'matrix_a': self.datatypes[0].values[0],
'matrix_b': self.datatypes[1].values[0],
@@ -72,8 +72,8 @@ class ProblemConfig:
}
@property
def layout_map(self) -> dict[str, str]:
"""Get current layout selections as a key-value map."""
def layout_map(self) -> Dict[str, str]:
"""Get layout as a key-value map."""
return {
'matrix_a': self.layouts[0].values[0],
'matrix_b': self.layouts[1].values[0],
@@ -83,7 +83,7 @@ class ProblemConfig:
@dataclass
class TileConfig:
"""configuration class for tile parameter."""
"""Configuration class for tile parameter."""
tile_m: Union[EnumConfigParam, RangeConfigParam]
tile_n: Union[EnumConfigParam, RangeConfigParam]
tile_k: Union[EnumConfigParam, RangeConfigParam]
@@ -99,7 +99,7 @@ class TileConfig:
@dataclass
class TraitConfig:
"""configuration class for kernel traits."""
"""Configuration class for kernel traits."""
pipeline: EnumConfigParam
scheduler: EnumConfigParam
epilogue: EnumConfigParam