[CK_TILE] Add pooling to ckTileEngine part3

2026-04-20 14:59:17 +00:00 · 2025-12-09 11:59:37 +00:00
parent 990f13229f
commit 07c078d5ef
8 changed files with 2449 additions and 44 deletions
--- a/tile_engine/ops/pooling/configs/default_config.json
+++ b/tile_engine/ops/pooling/configs/default_config.json
@@ -0,0 +1,35 @@
+{
+    "block_config": {
+        "block_m": {
+            "values": [64, 128, 256]
+        },
+        "block_n": {
+            "values": [1]
+        },
+        "warp_m": {
+            "values": [1, 2]
+        },
+        "warp_n": {
+            "values": [1]
+        },
+        "thread_tile_m": {
+            "values": [1, 2, 4]
+        },
+        "thread_tile_n": {
+            "values": [1]
+        }
+    },
+    "trait_config": {
+        "output_index": {
+            "values": [true, false]
+        },
+        "propagate_nan": {
+            "values": [false]
+        },
+        "pool_dim": {
+            "values": [2, 3]
+        }
+    },
+    "k_block_per_cu": 1
+}
+