saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-14 20:48:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Linux build fix

Browse source
This commit is contained in:
Ryan Hill 2021-05-19 13:49:52 -07:00
parent 9ef92f352f
commit a73a811516
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/framework/provider_bridge_ort.cc
View file

@ -1103,7 +1103,7 @@ CUptiResult CUPTIAPI cuptiActivityGetNextRecord(uint8_t* buffer, size_t validBuf
CUptiResult CUPTIAPI cuptiActivityRegisterCallbacks(CUpti_BuffersCallbackRequestFunc funcBufferRequested, CUpti_BuffersCallbackCompleteFunc funcBufferCompleted) {
auto* info = onnxruntime::GetProviderInfo_CUDA();
if (info)
return CUptiResult(info->cuptiActivityRegisterCallbacks(funcBufferRequested, funcBufferCompleted));
return CUptiResult(info->cuptiActivityRegisterCallbacks(reinterpret_cast<void*>(funcBufferRequested), reinterpret_cast<void*>(funcBufferCompleted)));
return CUPTI_ERROR_NOT_SUPPORTED;
}