zishuo/onnxruntime_go
1
0
Fork 0
mirror of https://github.com/yalue/onnxruntime_go.git synced 2024-05-03 08:45:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix a compiler error on unix systems

Browse Source 
- Slightly adjust the comments and error handling when looking up
   CoreML functionality.
This commit is contained in:
yalue 2023-08-30 13:51:16 -04:00
parent 4254585834
commit fa5a8b286b
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
setup_env.go
View File

@ -43,16 +43,18 @@ func platformCleanup() error {
// Should only be called on Apple systems; looks up the CoreML provider
// function which should only be exported on apple onnxruntime dylib files.
// For now, this will never return an error; instead it will silently set the
// coreml provider function to a null pointer which will be detected at
// runtime.
func setAppendCoreMLFunctionPointer(libraryHandle unsafe.Pointer) error {
// This function name must match the name in coreml_provider_factory.h,
// which is provided in the onnxruntime release's include/ directory on for
// Apple platforms.
cFunctionName := C.CString("OrtSessionOptionsAppendExecutionProvider_CoreML")
fnName := "OrtSessionOptionsAppendExecutionProvider_CoreML"
cFunctionName := C.CString(fnName)
defer C.free(unsafe.Pointer(cFunctionName))
appendCoreMLProviderProc := C.dlsym(libraryHandle, cFunctionName)
if appendCoreMLProviderProc == nil {
msg := C.GoString(C.dlerror())
return fmt.Errorf("Error looking up %s: %s", fnName, msg)
}
C.SetCoreMLProviderFunctionPointer(appendCoreMLProviderProc)
return nil
}
@ -85,15 +87,14 @@ func platformInitializeEnvironment() error {
return fmt.Errorf("Error setting ORT API base: %d", tmp)
}
if (runtime.GOOS == "darwin") || (runtime.GOOS == "ios") {
e = setAppendCoreMLFunctionPointer(handle)
if e != nil {
// This shouldn't actually ever happen in the current
// implementation; instead the coreml function should remain NULL,
// which will cause AppendExecutionProviderCoreML to return an
// error at runtime.
C.dlclose(handle)
return fmt.Errorf("Error finding CoreML functionality: %w", e)
}
setAppendCoreMLFunctionPointer(handle)
// We'll silently ignore potential errors returned by
// setAppendCoreMLFunctionPointer (for now at least). Even though we're
// on Apple hardware, it's possible that the user will have compiled
// the onnxruntime library from source without CoreML support.
// A failure here will only leave the coreml function pointer as NULL
// in our C code, which will be detected and result in an error at
// runtime.
}
libraryHandle = handle
return nil