Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

remove exception of java interface

This commit is contained in:
xuanyue 2022-09-01 17:15:13 +08:00
parent 1d76144a99
commit 51e8fd9ff1
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mindspore/lite/java/src/main/java/com/mindspore/ModelParallelRunner.java
View File

@ -21,6 +21,7 @@ import com.mindspore.config.RunnerConfig;
import java.util.ArrayList;
import java.util.List;
import java.util.Collections;
import java.util.concurrent.locks.ReentrantReadWriteLock;
/**
@ -96,11 +97,10 @@ public class ModelParallelRunner {
*/
public boolean predict(List<MSTensor> inputs, List<MSTensor> outputs) {
rwLock.readLock().lock();
if (this.modelParallelRunnerPtr == 0L) {
rwLock.readLock().unlock();
throw new IllegalStateException("predict cannot be called after calling free");
}
try {
if (this.modelParallelRunnerPtr == 0L) {
return false;
}
if (inputs == null || outputs == null || inputs.size() == 0) {
return false;
}
@ -148,7 +148,7 @@ public class ModelParallelRunner {
rwLock.readLock().lock();
if (this.modelParallelRunnerPtr == 0L) {
rwLock.readLock().unlock();
throw new IllegalStateException("getInputs cannot be called after calling free");
return Collections.emptyList();
}
List<Long> ret = this.getInputs(this.modelParallelRunnerPtr);
List<MSTensor> tensors = new ArrayList<>();
@ -169,7 +169,7 @@ public class ModelParallelRunner {
rwLock.readLock().lock();
if (this.modelParallelRunnerPtr == 0L) {
rwLock.readLock().unlock();
throw new IllegalStateException("getOutputs cannot be called after calling free");
return Collections.emptyList();
}
List<Long> ret = this.getOutputs(this.modelParallelRunnerPtr);
List<MSTensor> tensors = new ArrayList<>();