Skip to content

Commit 019abe1

Browse files
tlrmchlsmthabcdabcd987
authored andcommitted
Fix compilation error on CUDA 12.4
1 parent 9d477c9 commit 019abe1

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

csrc/all_to_all/internode_combine.cu

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -198,17 +198,17 @@ void AllToAllInterNode::combine(
198198
switch (splitMode) {
199199
case SplitMode::SEND:
200200
CUDACHECK(cudaLaunchCooperativeKernel(
201-
&combineKernel<T, NUM_WARPS, true, false>, dimGrid, dimBlock, args, 0, stream
201+
(void *)&combineKernel<T, NUM_WARPS, true, false>, dimGrid, dimBlock, args, 0, stream
202202
));
203203
break;
204204
case SplitMode::RECV:
205205
CUDACHECK(cudaLaunchCooperativeKernel(
206-
&combineKernel<T, NUM_WARPS, false, true>, dimGrid, dimBlock, args, 0, stream
206+
(void *)&combineKernel<T, NUM_WARPS, false, true>, dimGrid, dimBlock, args, 0, stream
207207
));
208208
break;
209209
case SplitMode::NONE:
210210
CUDACHECK(cudaLaunchCooperativeKernel(
211-
&combineKernel<T, NUM_WARPS, true, true>, dimGrid, dimBlock, args, 0, stream
211+
(void *)&combineKernel<T, NUM_WARPS, true, true>, dimGrid, dimBlock, args, 0, stream
212212
));
213213
break;
214214
default:

csrc/all_to_all/internode_dispatch.cu

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -303,7 +303,7 @@ void AllToAllInterNode::dispatch(
303303
switch (splitMode) {
304304
case SplitMode::SEND:
305305
CUDACHECK(cudaLaunchKernel(
306-
&dispatchKernel<NUM_WARPS, true, false>,
306+
(void *)&dispatchKernel<NUM_WARPS, true, false>,
307307
dimGrid,
308308
dimBlock,
309309
args,
@@ -313,12 +313,12 @@ void AllToAllInterNode::dispatch(
313313
break;
314314
case SplitMode::RECV:
315315
CUDACHECK(cudaLaunchCooperativeKernel(
316-
&dispatchKernel<NUM_WARPS, false, true>, dimGrid, dimBlock, args, 0, stream
316+
(void *)&dispatchKernel<NUM_WARPS, false, true>, dimGrid, dimBlock, args, 0, stream
317317
));
318318
break;
319319
case SplitMode::NONE:
320320
CUDACHECK(cudaLaunchCooperativeKernel(
321-
&dispatchKernel<NUM_WARPS, true, true>,
321+
(void *)&dispatchKernel<NUM_WARPS, true, true>,
322322
dimGrid,
323323
dimBlock,
324324
args,

0 commit comments

Comments
 (0)