Ck tile engine gemm (#2982)

* Partial Progress : CK Tile Engine GEMM

* Partial Progress : CK Tile Engine GEMM

* Partial Progress : Working GEMM Code

* Partial Progress : Working GEMM Code

* Changinf jenkins to remove preshuffle

* Partial Progress : CK TILE ENGINE GEMM Debugging

* Partial Progress : Removing changes that are not GEMM

* Partial Progress : Validation of full block size in GEMM

* Changes in Jenkins to run only fp16 and bf16

* Addressing Review Comments

* Partial Progress : Addressing CI issues

* Partial Progress - Runing GEMM for fp16,bf16 and rcr

* Clang

* Adding fp8 and bf8

* Adding fp8 and bf8

* Adding additional architrcture

* Limited datatypes and layouts

* Adding k_block_per_cu in test config

* Changes to faling CI errors

* Changes to faling CI errors

* Validation for GEMM

* Adding Layout support

* Adding Validations

* Adding layout in jenkins

* Update on Jenkins

* Distribution validation for GEMM

* Resolving merge conflicts

* Solving merge conflicts
This commit is contained in:
Thrupti Raj Lakshmana Gowda
2025-10-27 21:11:13 -05:00
committed by GitHub
parent b11f53a484
commit 7fc0a38e90
18 changed files with 504 additions and 987 deletions

View File

@@ -1,31 +1,28 @@
{
"problem": {
},
"tile_config": {
"tile_m": {
"values": [
128,
256
64
]
},
"tile_n": {
"values": [
128
192
]
},
"tile_k": {
"values": [
128
64
]
},
"warp_m": {
"values": [
4
2
]
},
"warp_n": {
"values": [
1
2
]
},
"warp_k": {
@@ -35,36 +32,33 @@
},
"warp_tile_m": {
"values": [
16, 32
32
]
},
"warp_tile_n": {
"values": [
16, 32
32
]
},
"warp_tile_k": {
"values": [
16, 32
8
]
}
},
"trait_config": {
"pipeline": {
"values": [
"compv3",
"mem"
"compv4"
]
},
"scheduler": {
"values": [
"intrawave",
"interwave"
"intrawave"
]
},
"epilogue": {
"values": [
"default",
"cshuffle"
]
},
@@ -85,9 +79,9 @@
},
"persistent": {
"values": [
false,
true
]
}
}
},
"k_block_per_cu": 1
}