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

Merge pull request #220 from Peter9606/wrong-stride-array-definition

Browse Source 
Bugfix: typo, make reduction device cases passed
This commit is contained in:
Haicheng Wu
2021-04-02 08:43:52 -04:00
committed by GitHub
parent b0bbc6d548 7074047a54
commit d97214987a
1 changed files with 1 additions and 1 deletions
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) {
int64_t src_stride[3];
int64_t dst_stride[2];
int64_t dst_stride[3];
switch (reduction_index) {
case 0: