squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 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
commit d97214987a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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: