[CK TILE ENGINE] Code changes to finding GPU id from TARGET (#3055)

* Reading gpuname from target for gemm in ck tile engine

* Reading gpuname from target for gemm preshuffle in ck tile engine

* Reading gpuname from target for gemm preshuffle in ck tile engine

* Get GPU changes for GEMM Muti D in TILE ENGINE

* Addressing errors for gpu name in cktileengine
This commit is contained in:
Thrupti Raj Lakshmana Gowda
2025-10-20 11:02:18 -05:00
committed by GitHub
parent f18b79f328
commit 9f77061094
12 changed files with 59 additions and 149 deletions

View File

@@ -15,8 +15,9 @@ logging.basicConfig(level=logging.INFO)
class GemmKernelBuilder:
def __init__(self, working_path, datatype, layout, config_json=None):
def __init__(self, working_path, gpu_target, datatype, layout, config_json=None):
self.working_path = Path(working_path)
self.gpu_target = gpu_target
self.datatype = datatype
self.layout = layout
self.config_json = config_json
@@ -231,6 +232,7 @@ class GemmKernelBuilder:
b_datatype,
c_datatype,
pipeline,
self.gpu_target,
)
def _generate_trait_combinations(self):
@@ -822,6 +824,11 @@ def main():
description="GEMM kernel instance builder with parallel support"
)
parser.add_argument("--working_path", required=True, help="Working directory path")
parser.add_argument(
"--gpu_target",
required=True,
help="GPU target architecture",
)
parser.add_argument(
"--datatype",
required=True,
@@ -861,7 +868,7 @@ def main():
# Create builder
builder = GemmKernelBuilder(
args.working_path, args.datatype, args.layout, args.config_json
args.working_path, args.gpu_target, args.datatype, args.layout, args.config_json
)
if args.list_kernels: