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

fix type hint (#2254)

Browse Source 
[ROCm/composable_kernel commit: 4286eae09a]
This commit is contained in:
Casey-Shi
2025-05-28 23:43:58 +08:00
committed by GitHub
parent ab12990991
commit f2fe8424ce
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