@mineproness/tensorloop 2.2.3 → 2.2.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mineproness/tensorloop",
-  "version": "2.2.3",
+  "version": "2.2.5",
   "description": "Raw-JAVASCRIPT Based Matirx and AI LLM Package called Tensorloop.",
   "keywords": [
     "ai",

package/src/Softmax/index.js

@@ -113,16 +113,16 @@ export class Softmax {
     }
     async crossEntropy(probs, target) {
         const encoder = this.device.createCommandEncoder()
-
         const tokenCount = probs.size / (this.vocabSize * 4)
         const targetBuffer = await this.gpu.createBuffer(target.length * Float32Array.BYTES_PER_ELEMENT)
         if(!this.loss) this.loss = await this.gpu.createBuffer(4 * tokenCount)
         if(!this.acc) this.acc = await this.gpu.createBuffer(4 * 1)
         await this.gpu.WriteBuffer(targetBuffer, new Uint32Array(target))
-        const bindGroup1 = await this.gpu.createBindGroup(this.lossPipline, 0, [probs.mainBuffer, targetBuffer, this.loss])
-        const bindGroup2 = await this.gpu.createBindGroup(this.accPipline, 0 , [this.paramsForward, probs.mainBuffer, targetBuffer, this.acc])
+        const bindGroup1 = await this.gpu.createBindGroup(this.lossPipline, 0, [probs.mainBuffer, targetBuffer, this.loss, this.paramsForward])
+        const bindGroup2 = await this.gpu.createBindGroup(this.accPipline, 0 , [this.paramsForward, probs.mainBuffer, this.acc, targetBuffer])
         await this.gpu.RunPipline(encoder, this.lossPipline, bindGroup1, Math.ceil((tokenCount / 4)))
         await this.gpu.RunPipline(encoder, this.accPipline, bindGroup2, 1)
+        await this.gpu.submitQueue(encoder)
         return {loss: await this.gpu.readBuffer(this.loss.size, this.loss), acc: (await this.gpu.readBuffer(this.acc.size, this.acc))[0]}
     }