From a267d47f9b24b367a68c68453fa7d317fa745662 Mon Sep 17 00:00:00 2001
From: Tarun Paparaju <tarun.paparaju@gmail.com>
Date: Sun, 10 Aug 2025 19:42:21 -0700
Subject: [PATCH] Update batched_gemm.cu (#2538)

---
 examples/05_batched_gemm/batched_gemm.cu | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/05_batched_gemm/batched_gemm.cu b/examples/05_batched_gemm/batched_gemm.cu
index 8b8d64c2..ea94eb39 100644
--- a/examples/05_batched_gemm/batched_gemm.cu
+++ b/examples/05_batched_gemm/batched_gemm.cu
@@ -243,10 +243,11 @@ cudaError_t run_batched_gemm(bool use_array) {
   const char* gemm_desc = use_array ? "array" : "strided batched";
   std::cout << "Running " << gemm_desc << " gemm" << std::endl;
 
-  // Arbitrary problem size
+  // Arbitrary matrix shape
   int const m = 520;
   int const n = 219;
   int const k = 129;
+
   int const batch_count = 17;
 
   // A, B are non-transpose, column major