From 3b34c090332aba0c03e3ad338a0f883e528a6fa1 Mon Sep 17 00:00:00 2001
From: leimao <dukeleimao@gmail.com>
Date: Thu, 9 Jan 2025 09:33:16 -0800
Subject: [PATCH] Use Compile-Time Constexpr If

---
 examples/cute/tutorial/sgemm_sm80.cu | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/cute/tutorial/sgemm_sm80.cu b/examples/cute/tutorial/sgemm_sm80.cu
index 5ae0bf0f8b..6c200ecc4c 100644
--- a/examples/cute/tutorial/sgemm_sm80.cu
+++ b/examples/cute/tutorial/sgemm_sm80.cu
@@ -208,10 +208,10 @@ gemm_device(ProblemShape shape_MNK, CtaTiler cta_tiler,
   Tensor tCsB_p = tCsB(_,_,_,smem_pipe_read);
 
   // Size of the register pipeline
-  auto K_BLOCK_MAX = size<2>(tCrA);
+  constexpr auto K_BLOCK_MAX = size<2>(tCrA);
 
   // PREFETCH register pipeline
-  if (K_BLOCK_MAX > 1) {
+  if constexpr (K_BLOCK_MAX > 1) {
     // Wait until our first prefetched tile is loaded in
     cp_async_wait<K_PIPE_MAX-2>();
     __syncthreads();