diff --git a/examples/05_bmg_gemm_with_epilogues/05_bmg_gemm_with_epilogue_splitk.cpp b/examples/05_bmg_gemm_with_epilogues/05_bmg_gemm_with_epilogue_splitk.cpp
index c4570ec3f3..461da9ac3b 100644
--- a/examples/05_bmg_gemm_with_epilogues/05_bmg_gemm_with_epilogue_splitk.cpp
+++ b/examples/05_bmg_gemm_with_epilogues/05_bmg_gemm_with_epilogue_splitk.cpp
@@ -218,7 +218,10 @@ struct ExampleRunner {
     auto D2 = std::vector<ElementOutput>(size(D2_shape));
     compat::memcpy<ElementOutput>(D.data(), block_ref_D.get(), size(D_shape));
     compat::wait();
-
+    if (N < NUM_HEAD * (NOPE_DIM + ROPE_DIM)) {
+      std::cout << "Error: n < num_head * (nope_dim + rope_dim). Please set a sufficiently large value for n. Skipping the check." << std::endl;
+      return true;
+    }
     for (int l = 0; l < L; l++) {
       for (int i = 0; i < M; i++) {
         for (int j = 0; j < NUM_HEAD; j++) {