Initial commit for Qwen2TornadoVMLayerPlanner

Author: orionpapadakis

src/main/java/org/beehive/gpullama3/tornadovm/Qwen2TornadoVMLayerPlanner.java

@@ -0,0 +1,46 @@
+package org.beehive.gpullama3.tornadovm;
+
+import org.beehive.gpullama3.auxiliary.Tuple2;
+import org.beehive.gpullama3.inference.state.Qwen2State;
+import org.beehive.gpullama3.inference.weights.tornado.Qwen2TornadoWeights;
+import org.beehive.gpullama3.model.Model;
+import org.beehive.gpullama3.model.qwen2.Qwen2Configuration;
+import uk.ac.manchester.tornado.api.GridScheduler;
+import uk.ac.manchester.tornado.api.ImmutableTaskGraph;
+import uk.ac.manchester.tornado.api.TaskGraph;
+
+import java.util.List;
+
+public class Qwen2TornadoVMLayerPlanner extends TornadoVMLayerPlanner<Qwen2State, Qwen2Configuration, Qwen2TornadoWeights> {
+
+    /**
+     * Constructs a TornadoVMLayerPlanner for the given Llama model.
+     *
+     * @param state
+     *         The state object containing model tensors and buffers
+     * @param model
+     *         The Llama model instance containing configuration and weights
+     */
+    public Qwen2TornadoVMLayerPlanner(Qwen2State state, Model model) {
+        super(state, model);
+    }
+
+    @Override
+    protected TaskGraph configureLayerDataTransfers(TaskGraph unifiedLayer, int layerIndex) {
+        throw new UnsupportedOperationException("configureLayerDataTransfers Not supported yet.");
+    }
+
+    @Override
+    public Tuple2<List<ImmutableTaskGraph>, GridScheduler> setupTornadoForwardPlanLayered() {
+        throw new UnsupportedOperationException("setupTornadoForwardPlanLayered Not supported yet.");
+    }
+
+    @Override
+    public Tuple2<List<ImmutableTaskGraph>, GridScheduler> setupTornadoForwardPlanLayeredNonNvidia() {
+        return setupTornadoForwardPlanLayered();
+    }
+
+    private GridScheduler setupQwen2GridSchedulersLayeredNonNvidia() {
+        throw new UnsupportedOperationException("setupQwen2GridSchedulersLayeredNonNvidia Not supported yet.");
+    }
+}

src/main/java/org/beehive/gpullama3/tornadovm/TornadoVMMasterPlan.java

@@ -2,6 +2,7 @@
 
 import org.beehive.gpullama3.auxiliary.Tuple2;
 import org.beehive.gpullama3.inference.state.Phi3State;
+import org.beehive.gpullama3.inference.state.Qwen2State;
 import org.beehive.gpullama3.inference.state.Qwen3State;
 import org.beehive.gpullama3.inference.state.State;
 import org.beehive.gpullama3.model.Configuration;
@@ -99,7 +100,7 @@ TornadoVMLayerPlanner createPlanner(State state, Model model) {
         return switch (model.getModelType()) {
             case LLAMA_3, MISTRAL -> new TornadoVMLayerPlanner(state, model);
             case PHI_3 -> new Phi3TornadoVMLayerPlanner((Phi3State) state, model);
-            case QWEN_2, DEEPSEEK_R1_DISTILL_QWEN -> throw new UnsupportedOperationException("TornadoVM QWEN 2 not supported");
+            case QWEN_2, DEEPSEEK_R1_DISTILL_QWEN -> new Qwen2TornadoVMLayerPlanner((Qwen2State) state, model);
             case QWEN_3 -> new Qwen3TornadoVMLayerPlanner((Qwen3State) state, model);
             case UNKNOWN -> throw new UnsupportedOperationException("Unknown model type");
         };