youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 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 12:54:44 -08: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

4
include/cutlass/detail/collective/mixed_input_utils.hpp
View File

@ -836,7 +836,7 @@ public:
}
}
else {
auto stage = make_tensor_like<ElementScale>(src_vm);
auto stage = make_tensor_like<ElementScale>(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 @@ public:
}
}
else {
auto stage = make_tensor_like<ElementScale>(src_vm);
auto stage = make_tensor_like<ElementScale>(src_vm(_, 0));
CUTLASS_PRAGMA_UNROLL
for (int i = 0; i < size<1>(dst_vm); ++i) {
LayoutAwareConvert(src_vm(_, i), stage);