From 2b6cfd34d166f9756cdddc8e0500f69bbac57aa9 Mon Sep 17 00:00:00 2001 From: Lain <28486541+IwakuraRein@users.noreply.github.com> Date: Tue, 10 Dec 2024 12:54:44 -0800 Subject: [PATCH] fix a typo that fails the compiling when ElementScale is not the same as MmaType (#1977) --- include/cutlass/detail/collective/mixed_input_utils.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/include/cutlass/detail/collective/mixed_input_utils.hpp b/include/cutlass/detail/collective/mixed_input_utils.hpp index c78cf4ba5..f9f348b9f 100644 --- a/include/cutlass/detail/collective/mixed_input_utils.hpp +++ b/include/cutlass/detail/collective/mixed_input_utils.hpp @@ -836,7 +836,7 @@ struct MixedInputUtils { } } else { - auto stage = make_tensor_like(src_vm); + auto stage = make_tensor_like(src_vm(_, 0)); CUTLASS_PRAGMA_UNROLL for (int i = 0; i < size<1>(dst_vm); ++i) { LayoutAwareConvert(src_vm(_, i), stage); @@ -868,7 +868,7 @@ struct MixedInputUtils { } } else { - auto stage = make_tensor_like(src_vm); + auto stage = make_tensor_like(src_vm(_, 0)); CUTLASS_PRAGMA_UNROLL for (int i = 0; i < size<1>(dst_vm); ++i) { LayoutAwareConvert(src_vm(_, i), stage);