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

[CK TILE ENGINE] CI configuration with basic cases (#3475)

Browse Source 
* [CK TILE ENGINE] Adding GEMM BASIC TEST in Kenkins

* fix RUN_TILE_ENGINE_BASIC_TESTS name typo

* [CK Tile Engine] Updating basic CI

* Resolving merging issues

* Resolving merging issues

---------

Co-authored-by: illsilin_amdeng <Illia.Silin@amd.com>
This commit is contained in:
Thrupti Raj Lakshmana Gowda
2025-12-24 12:45:56 -06:00
committed by GitHub
parent 7f68f3c4fa
commit 62a8ec155f
4 changed files with 311 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

89
tile_engine/ops/gemm/gemm_multi_d/configs/default_ci_config.json Normal file
View File

@@ -0,0 +1,89 @@
{
"tile_config": {
"tile_m": {
"max": 256,
"min": 64,
"step": 256
},
"tile_n": {
"max": 256,
"min": 64,
"step": 256
},
"tile_k": {
"max": 256,
"min": 64,
"step": 256
},
"warp_m": {
"values": [
2
]
},
"warp_n": {
"values": [
2
]
},
"warp_k": {
"values": [
1
]
},
"warp_tile_m": {
"values": [
32
]
},
"warp_tile_n": {
"values": [
32
]
},
"warp_tile_k": {
"values": [
64
]
}
},
"trait_config": {
"pipeline": {
"values": [
"compv4"
]
},
"scheduler": {
"values": [
"intrawave",
"interwave"
]
},
"epilogue": {
"values": [
"cshuffle"
]
},
"pad_m": {
"values": [
false
]
},
"pad_n": {
"values": [
false
]
},
"pad_k": {
"values": [
false
]
},
"persistent": {
"values": [
false,
true
]
}
},
"k_block_per_cu": 1
}

89
tile_engine/ops/gemm/gemm_preshuffle/configs/default_ci_config.json Normal file
View File

@@ -0,0 +1,89 @@
{
"tile_config": {
"tile_m": {
"max": 256,
"min": 64,
"step": 256
},
"tile_n": {
"max": 256,
"min": 64,
"step": 256
},
"tile_k": {
"max": 256,
"min": 64,
"step": 256
},
"warp_m": {
"values": [
2
]
},
"warp_n": {
"values": [
2
]
},
"warp_k": {
"values": [
1
]
},
"warp_tile_m": {
"values": [
16
]
},
"warp_tile_n": {
"values": [
16
]
},
"warp_tile_k": {
"values": [
32
]
}
},
"trait_config": {
"pipeline": {
"values": [
"preshufflev2"
]
},
"scheduler": {
"values": [
"default"
]
},
"epilogue": {
"values": [
"cshuffle"
]
},
"pad_m": {
"values": [
false
]
},
"pad_n": {
"values": [
false
]
},
"pad_k": {
"values": [
false
]
},
"persistent": {
"values": [
true,
false
]
}
},
"k_block_per_cu": 1,
"permute_n": true
}

89
tile_engine/ops/gemm/gemm_universal/configs/default_ci_config.json Normal file
View File

@@ -0,0 +1,89 @@
{
"tile_config": {
"tile_m": {
"max": 256,
"min": 64,
"step": 256
},
"tile_n": {
"max": 256,
"min": 64,
"step": 256
},
"tile_k": {
"max": 256,
"min": 64,
"step": 256
},
"warp_m": {
"values": [
2
]
},
"warp_n": {
"values": [
2
]
},
"warp_k": {
"values": [
1
]
},
"warp_tile_m": {
"values": [
16
]
},
"warp_tile_n": {
"values": [
16
]
},
"warp_tile_k": {
"values": [
32
]
}
},
"trait_config": {
"pipeline": {
"values": [
"compv4"
]
},
"scheduler": {
"values": [
"intrawave",
"interwave"
]
},
"epilogue": {
"values": [
"cshuffle"
]
},
"pad_m": {
"values": [
false
]
},
"pad_n": {
"values": [
false
]
},
"pad_k": {
"values": [
false
]
},
"persistent": {
"values": [
false,
true
]
}
},
"k_block_per_cu": 1
}