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

Renaming deviceNum to deviceId.

Browse source
This commit is contained in:
Adam Pocock 2020-04-14 16:50:17 -04:00 committed by Changming Sun
parent e9dc8954ac
commit bc9a199b16
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
java/src/main/java/ai/onnxruntime/OrtSession.java
View file

@ -551,11 +551,11 @@ public class OrtSession implements AutoCloseable {
/**
* Adds DirectML as an execution backend.
*
* @param deviceNum The id of the DirectML device.
* @param deviceId The id of the DirectML device.
* @throws OrtException If there was an error in native code.
*/
public void addDirectML(int deviceNum) throws OrtException {
addDirectML(OnnxRuntime.ortApiHandle, nativeHandle, deviceNum);
public void addDirectML(int deviceId) throws OrtException {
addDirectML(OnnxRuntime.ortApiHandle, nativeHandle, deviceId);
}
/**
@ -622,7 +622,7 @@ public class OrtSession implements AutoCloseable {
long apiHandle, long nativeHandle, int allowUnalignedBuffers, String settings)
throws OrtException;
private native void addDirectML(long apiHandle, long nativeHandle, int deviceNum)
private native void addDirectML(long apiHandle, long nativeHandle, int deviceId)
throws OrtException;
private native void addACL(long apiHandle, long nativeHandle, int useArena) throws OrtException;