mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-04-20 14:59:17 +00:00
[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:
committed by
GitHub
parent
f18b79f328
commit
9f77061094
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user