From b5a0c5d23edc9794cafdc995d17b64cd245dda33 Mon Sep 17 00:00:00 2001
From: Loser Cheems <losercheems@gmail.com>
Date: Mon, 30 Jun 2025 22:01:27 +0800
Subject: [PATCH] Updates copy function defaults and clarifies comments

Changes default Clear_OOB_MN parameter to true for safer out-of-bounds handling.

Updates tensor dimension comments to use more descriptive MMA_M and MMA_N naming convention for better code clarity.
---
 csrc/src/utils.h | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/csrc/src/utils.h b/csrc/src/utils.h
index 79d91fa..2e2df1b 100644
--- a/csrc/src/utils.h
+++ b/csrc/src/utils.h
@@ -497,7 +497,7 @@ __forceinline__ __device__ void copy(
 
 ////////////////////////////////////////////////////////////////////////////////////////////////////
 
-template <bool Is_even_MN=true, bool Clear_OOB_MN=false,
+template <bool Is_even_MN=true, bool Clear_OOB_MN=true,
           typename TiledCopy, typename Engine0, typename Layout0, typename Engine1, typename Layout1,
           typename Engine2, typename Layout2>
 __forceinline__ __device__ void copy_ZOH(
@@ -505,11 +505,11 @@ __forceinline__ __device__ void copy_ZOH(
     Tensor<Engine1, Layout1> &D, Tensor<Engine2, Layout2> const &identity_MN,
     const int max_M=0, const int max_N=0
 ) {
-    CUTE_STATIC_ASSERT_V(rank(S) == Int<3>{});          // (MMA, M, N) 
-    CUTE_STATIC_ASSERT_V(rank(D) == Int<3>{});          // (MMA, M, N)
+    CUTE_STATIC_ASSERT_V(rank(S) == Int<3>{});          // (MMA, MMA_M, MMA_N) 
+    CUTE_STATIC_ASSERT_V(rank(D) == Int<3>{});          // (MMA, MMA_M, MMA_N)
     CUTE_STATIC_ASSERT_V(size<0>(S) == size<0>(D));     // MMA
-    CUTE_STATIC_ASSERT_V(size<1>(S) == size<1>(D));     // M
-    CUTE_STATIC_ASSERT_V(size<2>(S) == size<2>(D));     // N
+    CUTE_STATIC_ASSERT_V(size<1>(S) == size<1>(D));     // MMA_M
+    CUTE_STATIC_ASSERT_V(size<2>(S) == size<2>(D));     // MMA_N
     
     #pragma unroll
     for (int m = 0; m < size<1>(S); ++m) {