ffi.rb in onnxruntime-0.2.0

- old
+ new
@@ -14,60 +14,122 @@
 
     # enums
     TensorElementDataType = enum(:undefined, :float, :uint8, :int8, :uint16, :int16, :int32, :int64, :string, :bool, :float16, :double, :uint32, :uint64, :complex64, :complex128, :bfloat16)
     OnnxType = enum(:unknown, :tensor, :sequence, :map, :opaque, :sparsetensor)
 
-    # session
-    attach_function :OrtCreateEnv, %i[int string pointer], :pointer
-    attach_function :OrtCreateSession, %i[pointer string pointer pointer], :pointer
-    attach_function :OrtCreateSessionFromArray, %i[pointer pointer size_t pointer pointer], :pointer
-    attach_function :OrtRun, %i[pointer pointer pointer pointer size_t pointer size_t pointer], :pointer
-    attach_function :OrtCreateSessionOptions, %i[pointer], :pointer
-    attach_function :OrtSetSessionGraphOptimizationLevel, %i[pointer int], :pointer
-    attach_function :OrtSetSessionThreadPoolSize, %i[pointer int], :pointer
+    class Api < ::FFI::Struct
+      layout \
+        :CreateStatus, callback(%i[int string], :pointer),
+        :GetErrorCode, callback(%i[pointer], :pointer),
+        :GetErrorMessage, callback(%i[pointer], :string),
+        :CreateEnv, callback(%i[int string pointer], :pointer),
+        :CreateEnvWithCustomLogger, callback(%i[], :pointer),
+        :EnableTelemetryEvents, callback(%i[pointer], :pointer),
+        :DisableTelemetryEvents, callback(%i[pointer], :pointer),
+        :CreateSession, callback(%i[pointer string pointer pointer], :pointer),
+        :CreateSessionFromArray, callback(%i[pointer pointer size_t pointer pointer], :pointer),
+        :Run, callback(%i[pointer pointer pointer pointer size_t pointer size_t pointer], :pointer),
+        :CreateSessionOptions, callback(%i[pointer], :pointer),
+        :SetOptimizedModelFilePath, callback(%i[], :pointer),
+        :CloneSessionOptions, callback(%i[], :pointer),
+        :SetSessionExecutionMode, callback(%i[], :pointer),
+        :EnableProfiling, callback(%i[], :pointer),
+        :DisableProfiling, callback(%i[], :pointer),
+        :EnableMemPattern, callback(%i[], :pointer),
+        :DisableMemPattern, callback(%i[], :pointer),
+        :EnableCpuMemArena, callback(%i[], :pointer),
+        :DisableCpuMemArena, callback(%i[], :pointer),
+        :SetSessionLogId, callback(%i[], :pointer),
+        :SetSessionLogVerbosityLevel, callback(%i[], :pointer),
+        :SetSessionLogSeverityLevel, callback(%i[], :pointer),
+        :SetSessionGraphOptimizationLevel, callback(%i[], :pointer),
+        :SetIntraOpNumThreads, callback(%i[], :pointer),
+        :SetInterOpNumThreads, callback(%i[], :pointer),
+        :CreateCustomOpDomain, callback(%i[], :pointer),
+        :CustomOpDomain_Add, callback(%i[], :pointer),
+        :AddCustomOpDomain, callback(%i[], :pointer),
+        :RegisterCustomOpsLibrary, callback(%i[], :pointer),
+        :SessionGetInputCount, callback(%i[pointer pointer], :pointer),
+        :SessionGetOutputCount, callback(%i[pointer pointer], :pointer),
+        :SessionGetOverridableInitializerCount, callback(%i[], :pointer),
+        :SessionGetInputTypeInfo, callback(%i[pointer size_t pointer], :pointer),
+        :SessionGetOutputTypeInfo, callback(%i[pointer size_t pointer], :pointer),
+        :SessionGetOverridableInitializerTypeInfo, callback(%i[], :pointer),
+        :SessionGetInputName, callback(%i[pointer size_t pointer pointer], :pointer),
+        :SessionGetOutputName, callback(%i[pointer size_t pointer pointer], :pointer),
+        :SessionGetOverridableInitializerName, callback(%i[], :pointer),
+        :CreateRunOptions, callback(%i[], :pointer),
+        :RunOptionsSetRunLogVerbosityLevel, callback(%i[], :pointer),
+        :RunOptionsSetRunLogSeverityLevel, callback(%i[], :pointer),
+        :RunOptionsSetRunTag, callback(%i[], :pointer),
+        :RunOptionsGetRunLogVerbosityLevel, callback(%i[], :pointer),
+        :RunOptionsGetRunLogSeverityLevel, callback(%i[], :pointer),
+        :RunOptionsGetRunTag, callback(%i[], :pointer),
+        :RunOptionsSetTerminate, callback(%i[], :pointer),
+        :RunOptionsUnsetTerminate, callback(%i[], :pointer),
+        :CreateTensorAsOrtValue, callback(%i[pointer pointer size_t int pointer], :pointer),
+        :CreateTensorWithDataAsOrtValue, callback(%i[pointer pointer size_t pointer size_t int pointer], :pointer),
+        :IsTensor, callback(%i[], :pointer),
+        :GetTensorMutableData, callback(%i[pointer pointer], :pointer),
+        :FillStringTensor, callback(%i[pointer pointer size_t], :pointer),
+        :GetStringTensorDataLength, callback(%i[], :pointer),
+        :GetStringTensorContent, callback(%i[], :pointer),
+        :CastTypeInfoToTensorInfo, callback(%i[pointer pointer], :pointer),
+        :GetOnnxTypeFromTypeInfo, callback(%i[pointer pointer], :pointer),
+        :CreateTensorTypeAndShapeInfo, callback(%i[], :pointer),
+        :SetTensorElementType, callback(%i[], :pointer),
+        :SetDimensions, callback(%i[], :pointer),
+        :GetTensorElementType, callback(%i[pointer pointer], :pointer),
+        :GetDimensionsCount, callback(%i[pointer pointer], :pointer),
+        :GetDimensions, callback(%i[pointer pointer size_t], :pointer),
+        :GetSymbolicDimensions, callback(%i[], :pointer),
+        :GetTensorShapeElementCount, callback(%i[pointer pointer], :pointer),
+        :GetTensorTypeAndShape, callback(%i[pointer pointer], :pointer),
+        :GetTypeInfo, callback(%i[pointer pointer], :pointer),
+        :GetValueType, callback(%i[pointer pointer], :pointer),
+        :CreateMemoryInfo, callback(%i[], :pointer),
+        :CreateCpuMemoryInfo, callback(%i[int int pointer], :pointer),
+        :CompareMemoryInfo, callback(%i[], :pointer),
+        :MemoryInfoGetName, callback(%i[], :pointer),
+        :MemoryInfoGetId, callback(%i[], :pointer),
+        :MemoryInfoGetMemType, callback(%i[], :pointer),
+        :MemoryInfoGetType, callback(%i[], :pointer),
+        :AllocatorAlloc, callback(%i[], :pointer),
+        :AllocatorFree, callback(%i[], :pointer),
+        :AllocatorGetInfo, callback(%i[], :pointer),
+        :GetAllocatorWithDefaultOptions, callback(%i[pointer], :pointer),
+        :AddFreeDimensionOverride, callback(%i[], :pointer),
+        :GetValue, callback(%i[pointer int pointer pointer], :pointer),
+        :GetValueCount, callback(%i[pointer pointer], :pointer),
+        :CreateValue, callback(%i[], :pointer),
+        :CreateOpaqueValue, callback(%i[], :pointer),
+        :GetOpaqueValue, callback(%i[], :pointer),
+        :KernelInfoGetAttribute_float, callback(%i[], :pointer),
+        :KernelInfoGetAttribute_int64, callback(%i[], :pointer),
+        :KernelInfoGetAttribute_string, callback(%i[], :pointer),
+        :KernelContext_GetInputCount, callback(%i[], :pointer),
+        :KernelContext_GetOutputCount, callback(%i[], :pointer),
+        :KernelContext_GetInput, callback(%i[], :pointer),
+        :KernelContext_GetOutput, callback(%i[], :pointer),
+        :ReleaseEnv, callback(%i[pointer], :void),
+        :ReleaseStatus, callback(%i[pointer], :void),
+        :ReleaseMemoryInfo, callback(%i[pointer], :void),
+        :ReleaseSession, callback(%i[pointer], :void),
+        :ReleaseValue, callback(%i[pointer], :void),
+        :ReleaseRunOptions, callback(%i[pointer], :void),
+        :ReleaseTypeInfo, callback(%i[pointer], :void),
+        :ReleaseTensorTypeAndShapeInfo, callback(%i[pointer], :void),
+        :ReleaseSessionOptions, callback(%i[pointer], :void),
+        :ReleaseCustomOpDomain, callback(%i[pointer], :void)
+    end
 
-    # input and output
-    attach_function :OrtSessionGetInputCount, %i[pointer pointer], :pointer
-    attach_function :OrtSessionGetOutputCount, %i[pointer pointer], :pointer
-    attach_function :OrtSessionGetInputTypeInfo, %i[pointer size_t pointer], :pointer
-    attach_function :OrtSessionGetOutputTypeInfo, %i[pointer size_t pointer], :pointer
-    attach_function :OrtSessionGetInputName, %i[pointer size_t pointer pointer], :pointer
-    attach_function :OrtSessionGetOutputName, %i[pointer size_t pointer pointer], :pointer
+    class ApiBase < ::FFI::Struct
+      # use uint32 instead of uint32_t
+      # to prevent "unable to resolve type" error on Ubuntu
+      layout \
+        :GetApi, callback(%i[uint32], Api.by_ref),
+        :GetVersionString, callback(%i[], :string)
+    end
 
-    # tensor
-    attach_function :OrtCreateTensorAsOrtValue, %i[pointer pointer size_t int pointer], :pointer
-    attach_function :OrtCreateTensorWithDataAsOrtValue, %i[pointer pointer size_t pointer size_t int pointer], :pointer
-    attach_function :OrtGetTensorMutableData, %i[pointer pointer], :pointer
-    attach_function :OrtIsTensor, %i[pointer pointer], :pointer
-    attach_function :OrtFillStringTensor, %i[pointer pointer size_t], :pointer
-    attach_function :OrtCastTypeInfoToTensorInfo, %i[pointer pointer], :pointer
-    attach_function :OrtOnnxTypeFromTypeInfo, %i[pointer pointer], :pointer
-    attach_function :OrtGetTensorElementType, %i[pointer pointer], :pointer
-    attach_function :OrtGetDimensionsCount, %i[pointer pointer], :pointer
-    attach_function :OrtGetDimensions, %i[pointer pointer size_t], :pointer
-    attach_function :OrtGetTensorShapeElementCount, %i[pointer pointer], :pointer
-    attach_function :OrtGetTensorTypeAndShape, %i[pointer pointer], :pointer
-
-    # value
-    attach_function :OrtGetTypeInfo, %i[pointer pointer], :pointer
-    attach_function :OrtGetValueType, %i[pointer pointer], :pointer
-
-    # maps and sequences
-    attach_function :OrtGetValue, %i[pointer int pointer pointer], :pointer
-    attach_function :OrtGetValueCount, %i[pointer pointer], :pointer
-
-    # version
-    attach_function :OrtGetVersionString, %i[], :string
-
-    # error
-    attach_function :OrtGetErrorMessage, %i[pointer], :string
-
-    # allocator
-    attach_function :OrtCreateCpuAllocatorInfo, %i[int int pointer], :pointer
-    attach_function :OrtCreateDefaultAllocator, %i[pointer], :pointer
-
-    # release
-    attach_function :OrtReleaseEnv, %i[pointer], :pointer
-    attach_function :OrtReleaseTypeInfo, %i[pointer], :pointer
-    attach_function :OrtReleaseStatus, %i[pointer], :pointer
+    attach_function :OrtGetApiBase, %i[], ApiBase.by_ref
   end
 end