Remove unnecessary size field and constructor from TornadoTensor and refactor subclasses accordingly.

Author: orionpapadakis

src/main/java/org/beehive/gpullama3/tensor/tornado/FP16TornadoTensor.java

@@ -1,22 +1,25 @@
 package org.beehive.gpullama3.tensor.tornado;
 
 import org.beehive.gpullama3.tensor.GGMLType;
+import uk.ac.manchester.tornado.api.types.HalfFloat;
 import uk.ac.manchester.tornado.api.types.arrays.HalfFloatArray;
 
 import java.lang.foreign.MemorySegment;
 
 public class FP16TornadoTensor extends TornadoTensor {
-    private final HalfFloatArray values;
+    private final HalfFloatArray tornadoNativeArray;
 
-    public FP16TornadoTensor(int size, MemorySegment segment) {
-        super(size);
-        this.values = new HalfFloatArray(size);
-        this.values.getSegment().copyFrom(segment);
+    public FP16TornadoTensor(HalfFloatArray halfFloatArray) {
+        this.tornadoNativeArray = halfFloatArray;
+    }
+
+    public FP16TornadoTensor(MemorySegment segment) {
+        this.tornadoNativeArray = new HalfFloatArray(segment);
     }
 
     @Override
     public HalfFloatArray asHalfFloatArray() {
-        return values;
+        return tornadoNativeArray;
     }
 
     @Override

src/main/java/org/beehive/gpullama3/tensor/tornado/FP32TornadoTensor.java

@@ -6,22 +6,19 @@
 import java.lang.foreign.MemorySegment;
 
 public class FP32TornadoTensor extends TornadoTensor {
-    private final FloatArray values;
+    private final FloatArray tornadoNativeArray;
 
-    public FP32TornadoTensor(FloatArray values) {
-        super(values.getSize());
-        this.values = values;
+    public FP32TornadoTensor(FloatArray floatArray) {
+        this.tornadoNativeArray = floatArray;
     }
 
-    public FP32TornadoTensor(int size, MemorySegment segment) {
-        super(size);
-        this.values = new FloatArray(size);
-        this.values.getSegment().copyFrom(segment);
+    public FP32TornadoTensor(MemorySegment segment) {
+        this.tornadoNativeArray = new FloatArray(segment);
     }
 
     @Override
     public FloatArray asFloatArray() {
-        return values;
+        return tornadoNativeArray;
     }
 
     @Override

src/main/java/org/beehive/gpullama3/tensor/tornado/Q8_0TornadoTensor.java

@@ -17,8 +17,7 @@ public class Q8_0TornadoTensor extends TornadoTensor {
     private final Int8Array quants;       // Quantized int8 values
     private MemorySegment segment;
 
-    public Q8_0TornadoTensor(int size, HalfFloatArray scales, Int8Array quants, MemorySegment segment) {
-        super(size);
+    public Q8_0TornadoTensor(HalfFloatArray scales, Int8Array quants, MemorySegment segment) {
         this.scales = scales;
         this.quants = quants;
         this.segment = segment;

src/main/java/org/beehive/gpullama3/tensor/tornado/TornadoTensor.java

@@ -10,15 +10,6 @@
  * These tensors wrap TornadoVM native arrays for GPU execution.
  */
 public abstract class TornadoTensor {
-    protected final int size;
-
-    protected TornadoTensor(int size) {
-        this.size = size;
-    }
-
-    public int size() {
-        return size;
-    }
 
     public abstract GGMLType type();