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

Bugfix: typo, make reduction device cases passed

Browse Source 
Signed-off-by: Peter Han <fujun.han@iluvatar.ai>
This commit is contained in:
Peter Han 2021-04-02 09:35:23 +08:00
parent 8a3e4b8d02
commit 7074047a54
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/cutlass/reduction/device/tensor_reduce.h
View File

@ -170,7 +170,7 @@ struct TensorReduction {
cudaStream_t stream = nullptr) { cudaStream_t stream = nullptr) {
int64_t src_stride[3]; int64_t src_stride[3];
int64_t dst_stride[2]; int64_t dst_stride[3];
switch (reduction_index) { switch (reduction_index) {
case 0: case 0: