Skip to content

Commit 7a96da1

Browse files
committed
Refactor TornadoVMLayerPlanner and TornadoVMMasterPlan for improved code clarity and organization
1 parent 36a228a commit 7a96da1

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

src/main/java/com/example/tornadovm/TornadoVMLayerPlanner.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,6 @@
4141
*
4242
* @see TaskGraph
4343
* @see GridScheduler
44-
* @see Llama
4544
*/
4645
// @formatter:on
4746
public class TornadoVMLayerPlanner {

src/main/java/com/example/tornadovm/TornadoVMMasterPlan.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
package com.example.tornadovm;
22

33
import com.example.auxiliary.Tuple2;
4+
import com.example.loader.weights.State;
45
import com.example.model.Configuration;
56
import com.example.model.Model;
6-
import com.example.loader.weights.State;
77
import uk.ac.manchester.tornado.api.GridScheduler;
88
import uk.ac.manchester.tornado.api.ImmutableTaskGraph;
99
import uk.ac.manchester.tornado.api.TornadoExecutionPlan;
@@ -24,7 +24,9 @@ public class TornadoVMMasterPlan {
2424

2525
public TornadoVMMasterPlan(State state, Model model, boolean isNvidia) {
2626
TornadoVMLayerPlanner tornadoVMLayerPlanner = new TornadoVMLayerPlanner(state, model);
27-
Tuple2<List<ImmutableTaskGraph>, GridScheduler> tornadoVMPlan = isNvidia ? tornadoVMLayerPlanner.setupTornadoForwardPlanLayered() : tornadoVMLayerPlanner.setupTornadoForwardPlanLayeredNonNvidia();
27+
Tuple2<List<ImmutableTaskGraph>, GridScheduler> tornadoVMPlan = isNvidia
28+
? tornadoVMLayerPlanner.setupTornadoForwardPlanLayered()
29+
: tornadoVMLayerPlanner.setupTornadoForwardPlanLayeredNonNvidia();
2830
this.taskGraphs = tornadoVMPlan.getFirst();
2931
this.scheduler = tornadoVMPlan.getSecond();
3032
this.state = state;

0 commit comments

Comments
 (0)