File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -140,10 +140,15 @@ urCommandBufferReleaseExp(ur_exp_command_buffer_handle_t hCommandBuffer) {
140
140
UR_APIEXPORT ur_result_t UR_APICALL
141
141
urCommandBufferFinalizeExp (ur_exp_command_buffer_handle_t hCommandBuffer) {
142
142
try {
143
+ const unsigned long long flags = 0 ;
143
144
#if CUDA_VERSION >= 12000
144
145
UR_CHECK_ERROR (cuGraphInstantiate (&hCommandBuffer->CudaGraphExec ,
145
- hCommandBuffer->CudaGraph , 0 ));
146
+ hCommandBuffer->CudaGraph , flags));
147
+ #elif CUDA_VERSION >= 11040
148
+ UR_CHECK_ERROR (cuGraphInstantiateWithFlags (
149
+ &hCommandBuffer->CudaGraphExec , hCommandBuffer->CudaGraph , flags));
146
150
#else
151
+ // Cannot use flags
147
152
UR_CHECK_ERROR (cuGraphInstantiate (&hCommandBuffer->CudaGraphExec ,
148
153
hCommandBuffer->CudaGraph , nullptr ,
149
154
nullptr , 0 ));
You can’t perform that action at this time.
0 commit comments