Skip to content

Commit 90c3592

Browse files
Fix task graph name in grid scheduler setup
1 parent d258c9d commit 90c3592

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/main/java/org/beehive/gpullama3/tornadovm/layers/type/fp16/Qwen3FP16FFNLayers.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,7 @@ List<ImmutableTaskGraph> setupFFNLayered() {
172172
* Setup a single transformer layer for Qwen3 with GQA
173173
*/
174174
TaskGraph setupSingleQwen3FFNLayer(Qwen3TornadoWeights weights, int layerIndex) {
175-
var taskGraphName = "ffn_layer_" + layerIndex;
175+
var taskGraphName = "layer_" + layerIndex;
176176
TaskGraph unifiedLayer = new TaskGraph(taskGraphName);
177177
unifiedLayer.consumeFromDevice(state.wrapX);
178178
unifiedLayer.transferToDevice(DataTransferMode.FIRST_EXECUTION, //

0 commit comments

Comments
 (0)