Skip to content

Commit d6b8cc9

Browse files
authored
Merge pull request #41 from alibuild/alibot-cleanup-15571
Please consider the following formatting changes to #15571
2 parents 039f813 + 1023771 commit d6b8cc9

1 file changed

Lines changed: 7 additions & 6 deletions

File tree

GPU/GPUTracking/Base/cuda/GPUReconstructionCUDA.cu

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -631,7 +631,8 @@ void GPUReconstructionCUDA::loadKernelModules(bool perKernel)
631631
} \
632632
}
633633

634-
void GPUReconstructionCUDA::SetONNXGPUStream(Ort::SessionOptions& sessionOptions, int32_t stream, int32_t* deviceId) {
634+
void GPUReconstructionCUDA::SetONNXGPUStream(Ort::SessionOptions& sessionOptions, int32_t stream, int32_t* deviceId)
635+
{
635636
GPUChkErr(cudaGetDevice(deviceId));
636637

637638
#if !defined(__HIPCC__) && defined(ORT_CUDA_BUILD)
@@ -645,17 +646,17 @@ void GPUReconstructionCUDA::SetONNXGPUStream(Ort::SessionOptions& sessionOptions
645646
const char* keys[] = {"device_id", "trt_int8_enable"};
646647
const char* values[] = {device.c_str(), "1"};
647648

648-
ORTCHK(api->UpdateTensorRTProviderOptions(trtOptions,keys,values,sizeof(keys) / sizeof(keys[0])));
649-
ORTCHK(api->UpdateTensorRTProviderOptionsWithValue(trtOptions,"user_compute_stream",mInternals->Streams[stream]));
650-
ORTCHK(api->SessionOptionsAppendExecutionProvider_TensorRT_V2(sessionOptions,trtOptions)); // Register TensorRT first: it consequently has higher priority.
649+
ORTCHK(api->UpdateTensorRTProviderOptions(trtOptions, keys, values, sizeof(keys) / sizeof(keys[0])));
650+
ORTCHK(api->UpdateTensorRTProviderOptionsWithValue(trtOptions, "user_compute_stream", mInternals->Streams[stream]));
651+
ORTCHK(api->SessionOptionsAppendExecutionProvider_TensorRT_V2(sessionOptions, trtOptions)); // Register TensorRT first: it consequently has higher priority.
651652
api->ReleaseTensorRTProviderOptions(trtOptions);
652653
#endif
653654

654655
// CUDA is the fallback for nodes unsupported by TensorRT.
655656
OrtCUDAProviderOptionsV2* cudaOptions = nullptr;
656657
ORTCHK(api->CreateCUDAProviderOptions(&cudaOptions));
657-
ORTCHK(api->UpdateCUDAProviderOptionsWithValue(cudaOptions,"user_compute_stream",mInternals->Streams[stream]));
658-
ORTCHK(api->SessionOptionsAppendExecutionProvider_CUDA_V2(sessionOptions,cudaOptions));
658+
ORTCHK(api->UpdateCUDAProviderOptionsWithValue(cudaOptions, "user_compute_stream", mInternals->Streams[stream]));
659+
ORTCHK(api->SessionOptionsAppendExecutionProvider_CUDA_V2(sessionOptions, cudaOptions));
659660
api->ReleaseCUDAProviderOptions(cudaOptions);
660661

661662
#elif defined(ORT_ROCM_BUILD)

0 commit comments

Comments
 (0)