We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6132b64 commit cde76dcCopy full SHA for cde76dc
src/main/java/com/example/tornadovm/TornadoVMMasterPlan.java
@@ -105,11 +105,12 @@ public boolean shouldUseNvidiaScheduler(Model model) {
105
String platformName = runtime.getBackend(0).getDefaultDevice().getPlatformName().toLowerCase(Locale.ROOT);
106
107
boolean isNvidia = platformName.contains("nvidia");
108
- boolean isMistral = model.getModelType() == ModelType.MISTRAL;
+ boolean isNotMistral = model.getModelType() != ModelType.MISTRAL;
109
110
- return !isNvidia || isMistral;
111
- }
+ boolean result = isNvidia && isNotMistral;
112
+ return result;
113
+ }
114
115
/**
116
* Executes the forward pass of a LLaMA transformer model using TornadoVM acceleration.
0 commit comments