Update batched_gemm.cu (#2538)

2025-08-10 19:42:21 -07:00
parent 9e6ab77d27
commit a267d47f9b
1 changed files with 2 additions and 1 deletions
--- 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