From b1b62a9574b5f7128a625dd1491123e93f2ded45 Mon Sep 17 00:00:00 2001
From: mx-flaggems-user <m01080@metax-tech.com>
Date: Tue, 18 Mar 2025 10:01:19 +0800
Subject: [PATCH] [Metax] Add metax backend tune config

---
 .../runtime/backend/_metax/tune_configs.yaml  | 33 +++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/src/flag_gems/runtime/backend/_metax/tune_configs.yaml b/src/flag_gems/runtime/backend/_metax/tune_configs.yaml
index e61f44a7b..aa66bdb32 100644
--- a/src/flag_gems/runtime/backend/_metax/tune_configs.yaml
+++ b/src/flag_gems/runtime/backend/_metax/tune_configs.yaml
@@ -1148,3 +1148,36 @@ batch_norm:
   - 4
   - 8
   - 16
+kron:
+- gen: true
+  param_map:
+    META:
+      BLOCK_M: block_m
+      BLOCK_N: block_n
+    num_warps: warps
+  block_m:
+  - 2
+  - 4
+  - 8
+  block_n:
+  - 1024
+  - 2048
+  warps:
+  - 4
+  - 8
+  - 16
+
+index_put:
+- gen: true
+  param_map:
+    META:
+      BLOCK_SIZE0: block_size0
+      BLOCK_SIZE1: block_size1
+  block_size0:
+  - 1
+  - 2
+  - 4
+  block_size1:
+  - 1024
+  - 2048
+  - 4096