From 7074047a542f43f6450df987abc8897d62223de9 Mon Sep 17 00:00:00 2001 From: Peter Han Date: Fri, 2 Apr 2021 09:35:23 +0800 Subject: [PATCH] Bugfix: typo, make reduction device cases passed Signed-off-by: Peter Han --- include/cutlass/reduction/device/tensor_reduce.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/include/cutlass/reduction/device/tensor_reduce.h b/include/cutlass/reduction/device/tensor_reduce.h index c67b205e..f662b43c 100644 --- a/include/cutlass/reduction/device/tensor_reduce.h +++ b/include/cutlass/reduction/device/tensor_reduce.h @@ -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: