Skip to content

Commit dfad57a

Browse files
committed
Improves function parameter formatting for readability
Reformats parameter lists in template function declarations to use consistent line breaks and alignment. Enhances code readability and maintainability by organizing long parameter lists across multiple lines in a more structured manner.
1 parent f7459fe commit dfad57a

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

csrc/src/utils.h

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -425,9 +425,10 @@ template <bool Is_even_MN=true, bool Is_even_K=true, bool Clear_OOB_MN=false, bo
425425
typename TiledCopy, typename Engine0, typename Layout0, typename Engine1, typename Layout1,
426426
typename Engine2, typename Layout2, typename Engine3, typename Layout3>
427427
__forceinline__ __device__ void copy(
428-
TiledCopy tiled_copy, Tensor<Engine0, Layout0> const &S,
429-
Tensor<Engine1, Layout1> &D, Tensor<Engine2, Layout2> const &identity_MN,
430-
Tensor<Engine3, Layout3> const &predicate_K, const int max_MN=0
428+
TiledCopy tiled_copy,
429+
Tensor<Engine0, Layout0> const &S, Tensor<Engine1, Layout1> &D,
430+
Tensor<Engine2, Layout2> const &identity_MN, Tensor<Engine3, Layout3> const &predicate_K,
431+
const int max_MN=0
431432
) {
432433
CUTE_STATIC_ASSERT_V(rank(S) == Int<3>{});
433434
CUTE_STATIC_ASSERT_V(rank(D) == Int<3>{});
@@ -496,8 +497,9 @@ template <bool Is_even_MN=true, bool Clear_OOB_MN=true,
496497
typename TiledCopy, typename Engine0, typename Layout0, typename Engine1, typename Layout1,
497498
typename Engine2, typename Layout2>
498499
__forceinline__ __device__ void copy_MN(
499-
TiledCopy tiled_copy, Tensor<Engine0, Layout0> const &S,
500-
Tensor<Engine1, Layout1> &D, Tensor<Engine2, Layout2> const &identity_MN,
500+
TiledCopy tiled_copy,
501+
Tensor<Engine0, Layout0> const &S, Tensor<Engine1, Layout1> &D,
502+
Tensor<Engine2, Layout2> const &identity_MN,
501503
const int max_M=0, const int max_N=0
502504
) {
503505
CUTE_STATIC_ASSERT_V(rank(S) == Int<3>{}); // (MMA, MMA_M, MMA_N)

0 commit comments

Comments
 (0)