Small fix.

azchohfi · Oct 8, 2024 · 8fd5f39 · 8fd5f39
1 parent f7341c3
commit 8fd5f39
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/csharp/src/Microsoft.ML.OnnxRuntime/NativeMethods.shared.cs b/csharp/src/Microsoft.ML.OnnxRuntime/NativeMethods.shared.cs
@@ -2130,7 +2130,7 @@ out IntPtr lora_adapter
 
 #if !__MOBILE__
  [UnmanagedFunctionPointer(CallingConvention.Winapi)]
- public delegate IntPtr /*(OrtStatus*)*/ DOrtSessionOptionsAppendExecutionProvider_DML1(IntPtr /*(OrtSessionOptions*) */ options, IntPtr dml_device, IntPtr cmd_queue);
+ public delegate IntPtr /*(OrtStatus*)*/ DOrtSessionOptionsAppendExecutionProvider_DML1(IntPtr /*(OrtSessionOptions*) */ options, object dml_device, object cmd_queue);
 
  public static DOrtSessionOptionsAppendExecutionProvider_DML1 OrtSessionOptionsAppendExecutionProvider_DML1;
 #endif

diff --git a/csharp/src/Microsoft.ML.OnnxRuntime/SessionOptions.shared.cs b/csharp/src/Microsoft.ML.OnnxRuntime/SessionOptions.shared.cs
@@ -280,7 +280,7 @@ public void AppendExecutionProvider_DML(int deviceId = 0)
  /// </summary>
  /// <param name="dmlDevice">A IDMLDevice reference</param>
  /// <param name="commandQueue">A ID3D12CommandQueue reference</param>
- public void AppendExecutionProvider_DML1(IntPtr dmlDevice, IntPtr commandQueue)
+ public void AppendExecutionProvider_DML1(object dmlDevice, object commandQueue)
  {
 #if __MOBILE__
  throw new NotSupportedException("The DML Execution Provider is not supported in this build");