leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 25 Packages Projects Releases Wiki Activity

fix a typo that fails the compiling when ElementScale is not the same as MmaType (#1977)

Browse Source
This commit is contained in:
Lain
2024-12-10 15:54:44 -05:00
committed by GitHub
parent 4c42f73fda
commit 2b6cfd34d1
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/cutlass/detail/collective/mixed_input_utils.hpp
+2 -2
View File
@@ -836,7 +836,7 @@ public:
} }
} }
else { else {
auto stage = make_tensor_like<ElementScale>(src_vm); auto stage = make_tensor_like<ElementScale>(src_vm(_, 0));
CUTLASS_PRAGMA_UNROLL CUTLASS_PRAGMA_UNROLL
for (int i = 0; i < size<1>(dst_vm); ++i) { for (int i = 0; i < size<1>(dst_vm); ++i) {
LayoutAwareConvert(src_vm(_, i), stage); LayoutAwareConvert(src_vm(_, i), stage);
@@ -868,7 +868,7 @@ public:
} }
} }
else { else {
auto stage = make_tensor_like<ElementScale>(src_vm); auto stage = make_tensor_like<ElementScale>(src_vm(_, 0));
CUTLASS_PRAGMA_UNROLL CUTLASS_PRAGMA_UNROLL
for (int i = 0; i < size<1>(dst_vm); ++i) { for (int i = 0; i < size<1>(dst_vm); ++i) {
LayoutAwareConvert(src_vm(_, i), stage); LayoutAwareConvert(src_vm(_, i), stage);