Skip to content

Commit 610027d

Browse files
fix the version guard
Signed-off-by: Youngeun Kwon <[email protected]>
1 parent a0170cc commit 610027d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

transformer_engine/common/comm_gemm_overlap/userbuffers/userbuffers.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1366,7 +1366,7 @@ __global__ void __launch_bounds__(MAX_THREADS)
13661366
cfg.attrs = attribute_ub; \
13671367
cfg.numAttrs = comm->sm_arch >= 9 ? 2 : 1;
13681368

1369-
#if defined(__CUDA_ARCH__) && (__CUDA_ARCH__ >= 900) && (CUDART_VERSION >= 12030)
1369+
#if (CUDART_VERSION >= 12030)
13701370
#define ADD_LAUNCH_COMPLETION_EVENT(attribute_ub, comm_launch_event) \
13711371
attribute_ub[2].id = cudaLaunchAttributeLaunchCompletionEvent; \
13721372
attribute_ub[2].val.launchCompletionEvent.event = comm_launch_event;

0 commit comments

Comments
 (0)