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