@@ -51,17 +51,17 @@ protected TaskGraph configureLayerDataTransfers(TaskGraph unifiedLayer, int laye
5151 context , state .wrapXb , state .wrapXb2 , //
5252 state .wrapQ , state .wrapK , state .wrapV , //
5353 state .wrapKeyCache , state .wrapValueCache , //
54- state .wrapAtt , state .wrapHb ,
54+ state .wrapAtt , state .wrapHb ); // ,
5555 // dbg buffers
56- state .dbgQ , state .dbgKeyCache , state .dbgValueCache , state .dbgXb , state .dbgX ); //
56+ // state.dbgQ, state.dbgKeyCache, state.dbgValueCache, state.dbgXb, state.dbgX); //
5757 } else {
5858 // Subsequent layers: Consume data already on device from previous layer
5959 unifiedLayer .consumeFromDevice (context , state .wrapXb , state .wrapXb2 , //
6060 state .wrapQ , state .wrapK , state .wrapV , //
6161 state .wrapKeyCache , state .wrapValueCache , //
6262 state .wrapAtt , state .wrapHb , //
63- state .positionHolder , //
64- state .dbgQ , state .dbgKeyCache , state .dbgValueCache , state .dbgXb , state .dbgX
63+ state .positionHolder //
64+ // state.dbgQ, state.dbgKeyCache, state.dbgValueCache, state.dbgXb, state.dbgX
6565 );
6666 }
6767 return unifiedLayer ;
0 commit comments