From e9362082c3ebf6184dbbfab503a0c0e44026174c Mon Sep 17 00:00:00 2001
From: Djordje Antic <djoantic@amd.com>
Date: Wed, 17 Dec 2025 12:26:02 +0000
Subject: [PATCH] Add gemm_gemm

---
 tuna/rocmlir/config_type.py    | 1 +
 tuna/rocmlir/rocmlir_worker.py | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/tuna/rocmlir/config_type.py b/tuna/rocmlir/config_type.py
index f55fe4ac4..e4c965d21 100644
--- a/tuna/rocmlir/config_type.py
+++ b/tuna/rocmlir/config_type.py
@@ -33,6 +33,7 @@ class ConfigType(Enum):
   convolution: str = 'convolution'
   gemm: str = 'gemm'
   attention: str = 'attention'
+  gemm_gemm: str = 'gemm_gemm'
 
   def __str__(self) -> str:
     return self.value
diff --git a/tuna/rocmlir/rocmlir_worker.py b/tuna/rocmlir/rocmlir_worker.py
index 8c9ab951f..30e31fe67 100644
--- a/tuna/rocmlir/rocmlir_worker.py
+++ b/tuna/rocmlir/rocmlir_worker.py
@@ -180,6 +180,8 @@ def run_cmd(self):
       special_args = "--operation gemm"
     elif self.dbt.config_type == ConfigType.attention:
       special_args = "--operation attention --verify-mode none"
+    elif self.dbt.config_type == ConfigType.gemm_gemm:
+      special_args = "--operation gemm_gemm"
     else:
       raise ValueError(f"Config type {self.dbt.config_type} not yet supported.")
     if self.dbt.session.tuning_space: