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) {
140140UR_APIEXPORT ur_result_t UR_APICALL
141141urCommandBufferFinalizeExp (ur_exp_command_buffer_handle_t hCommandBuffer) {
142142 try {
143+ const unsigned long long flags = 0 ;
143144#if CUDA_VERSION >= 12000
144145 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));
146150#else
151+ // Cannot use flags
147152 UR_CHECK_ERROR (cuGraphInstantiate (&hCommandBuffer->CudaGraphExec ,
148153 hCommandBuffer->CudaGraph , nullptr ,
149154 nullptr , 0 ));
You can’t perform that action at this time.
0 commit comments