Skip to content

Commit d68d0b8

Browse files
issue/1031 fix sum iluvatar
1 parent 8d7e2f8 commit d68d0b8

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/infiniop/ops/sum/nvidia/sum_nvidia.cu

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,8 @@ infiniStatus_t Descriptor::calculate(
107107
CALCULATE_SUM_WITH_BLOCK_SIZE(CUDA_BLOCK_SIZE_1024)
108108
} else if (_opaque->internal->maxThreadsPerBlock() == CUDA_BLOCK_SIZE_512) {
109109
CALCULATE_SUM_WITH_BLOCK_SIZE(CUDA_BLOCK_SIZE_512)
110+
} else if (_opaque->internal->maxThreadsPerBlock() == CUDA_BLOCK_SIZE_2048) {
111+
CALCULATE_SUM_WITH_BLOCK_SIZE(CUDA_BLOCK_SIZE_2048)
110112
} else if (_opaque->internal->maxThreadsPerBlock() == CUDA_BLOCK_SIZE_4096) {
111113
CALCULATE_SUM_WITH_BLOCK_SIZE(CUDA_BLOCK_SIZE_4096)
112114
} else {

0 commit comments

Comments
 (0)