Skip to content

Commit d381aeb

Browse files
Add debug log for forwardTornadoVM execution
1 parent e970c6e commit d381aeb

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/main/java/org/beehive/gpullama3/inference/InferenceCore.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -705,6 +705,8 @@ public static FloatArray forwardTornadoVM(Model model, State state, int token, i
705705
final Configuration configuration = model.configuration();
706706
final TornadoWeights weights = (TornadoWeights) model.weights();
707707

708+
System.out.println("forwardTornadoVM");
709+
708710
switch (weights.getWeightType()) {
709711
case F16 -> {
710712
MemorySegment tokenEmbeddings = weights.getTokenEmbeddingTable().asHalfFloatArray().getSegment();

0 commit comments

Comments
 (0)