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

Adjusting code indentation (#1639)

Browse Source
This commit is contained in:
103yiran 2024-10-24 00:55:02 +08:00 committed by GitHub
parent e5f3caf145
commit 03e3bffaec
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
include/cutlass/epilogue/threadblock/epilogue.h
View File

@ -512,24 +512,24 @@ public:
shared_load_iterator_.add_pointer_offset(kSmemPointerOffset);
shared_load_iterator_.load(aligned_accum_fragment[i]);
aligned_accum_fragment[0] = add_fragments(aligned_accum_fragment[0], aligned_accum_fragment[i]);
}
shared_load_iterator_.add_pointer_offset((1 - kPartitionsK) * kSmemPointerOffset);
}
//
// Compute the output result
//
shared_load_iterator_.add_pointer_offset((1 - kPartitionsK) * kSmemPointerOffset);
}
typename OutputTileIterator::Fragment output_fragment;
source.apply_output_operator(output_fragment, output_op, aligned_accum_fragment[0]);
//
// Compute the output result
//
//
// Store the final result
//
typename OutputTileIterator::Fragment output_fragment;
source.apply_output_operator(output_fragment, output_op, aligned_accum_fragment[0]);
destination_iterator.store(output_fragment);
++destination_iterator;
//
// Store the final result
//
destination_iterator.store(output_fragment);
++destination_iterator;
}
}
};