ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-11 01:27:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Partial Progress : Working till Listing kernels

Browse Source
This commit is contained in:
ThruptiRajLakshmanaGowda
2025-11-20 18:23:37 +00:00
parent eeeff3fbfe
commit d6db805e82
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tile_engine/ops_new/gemm/gemm_instance_builder.py
View File

@@ -44,7 +44,7 @@ class GemmKernelBuilder:
with open(config_json, "r") as f:
self.config = json.load(f)
def write_kernel_list(self, kernel_name_prefix):
def list_kernels(self, kernel_name_prefix):
"""Write kernel list to file for CMake to read (with comprehensive validation)"""
# Get configurations using comprehensive validation
tile_configs = self._get_tile_configs(kernel_name_prefix)

3
tile_engine/ops_new/gemm/gemm_universal/gemm_universal_instance_builder.py
View File

@@ -96,8 +96,7 @@ def main():
)
if args.list_kernels:
builder.write_kernel_list("gemm_universal") # Rename this method name
# pass
builder.list_kernels("gemm_universal")
elif args.gen_single:
# # Generate a single kernel file
# if not args.kernel_name or not args.tile_config or not args.trait_combo: