@genai-fi/nanogpt 0.11.0 → 0.12.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/Generator.js +29 -29
- package/dist/{RealDiv-Ds-jvL09.js → RealDiv-C8neBwFi.js} +17 -17
- package/dist/{Reshape-Cd6e-Otn.js → Reshape-Bd4V_4X7.js} +1 -1
- package/dist/{Reshape-Ct266DEk.js → Reshape-Ck29jQSY.js} +7 -7
- package/dist/TeachableLLM.d.ts +2 -1
- package/dist/TeachableLLM.js +9 -9
- package/dist/Trainer.d.ts +4 -2
- package/dist/Trainer.js +12 -9
- package/dist/{axis_util-DofAuy0p.js → axis_util-DGqbT-FX.js} +1 -1
- package/dist/backend.js +2 -2
- package/dist/{backend_util-C7NWHpv7.js → backend_util-DC3rBo_H.js} +18 -18
- package/dist/{backend_webgpu-B0Vls736.js → backend_webgpu-mbhNnlx9.js} +10 -10
- package/dist/{broadcast_to-DDaNMbX7.js → broadcast_to-D1Dmg2Oz.js} +2 -2
- package/dist/checks/appendCache.js +2 -2
- package/dist/checks/attentionMask.js +3 -3
- package/dist/checks/gelu.js +2 -2
- package/dist/checks/matMulGelu.js +2 -2
- package/dist/checks/normRMS.js +4 -4
- package/dist/checks/normRMSGrad.js +3 -3
- package/dist/checks/packUnpack.js +2 -2
- package/dist/checks/qkv.js +2 -2
- package/dist/checks/rope.js +2 -2
- package/dist/clip_by_value-fg2aKzUy.js +12 -0
- package/dist/{complex-DClmWqJt.js → complex-Cyg-eQeZ.js} +1 -1
- package/dist/concat-CSm2rMwe.js +17 -0
- package/dist/{concat_util-CHsJFZJJ.js → concat_util-D0je5Ppu.js} +1 -1
- package/dist/{dataset-DcjWqUVQ.js → dataset-CVIJu7Xa.js} +3 -3
- package/dist/{dropout-OxuaJz6z.js → dropout-DLhSMNTZ.js} +14 -14
- package/dist/expand_dims-ChkuOp6I.js +11 -0
- package/dist/{exports_initializers-eS9QJ6ut.js → exports_initializers-1KWPiStI.js} +1 -1
- package/dist/{floor-DIb-lN_u.js → floor-BRMPgeIs.js} +1 -1
- package/dist/gather-BSULDalH.js +9 -0
- package/dist/{gelu-DqTbCx5x.js → gelu-BK1k-n1i.js} +1 -1
- package/dist/{gpgpu_math-CJcbnKPC.js → gpgpu_math-BJSTk_mW.js} +25 -25
- package/dist/{index-Dj5TkmPY.js → index-BBVLAXZD.js} +14 -14
- package/dist/{index-D0RBWjq8.js → index-Duu1Lvvv.js} +45 -45
- package/dist/{kernel_funcs_utils-CSaumNDs.js → kernel_funcs_utils-BtYrPoJu.js} +8 -8
- package/dist/layers/BaseLayer.js +2 -2
- package/dist/layers/CausalSelfAttention.js +6 -6
- package/dist/layers/MLP.js +4 -4
- package/dist/layers/PositionEmbedding.js +5 -5
- package/dist/layers/RMSNorm.js +3 -3
- package/dist/layers/RoPECache.js +4 -4
- package/dist/layers/TiedEmbedding.js +6 -6
- package/dist/layers/TransformerBlock.js +1 -1
- package/dist/loader/loadTransformers.js +1 -1
- package/dist/loader/oldZipLoad.js +17 -17
- package/dist/{log_sum_exp-VLZgbFAH.js → log_sum_exp-CVqLsVLl.js} +4 -4
- package/dist/main.d.ts +9 -0
- package/dist/main.js +69 -58
- package/dist/{matMul16-cDxwemKj.js → matMul16-xswmhSuF.js} +7 -7
- package/dist/{matMulGelu-B2s_80-H.js → matMulGelu-BpvgnYG8.js} +26 -26
- package/dist/mat_mul-Bn2BDpT4.js +11 -0
- package/dist/{mod-PrOKlFxH.js → mod-B4AUd1Np.js} +1 -1
- package/dist/models/NanoGPTV1.js +2 -2
- package/dist/models/model.js +9 -9
- package/dist/{ones-BX_wEgzB.js → ones-CBI1AQjb.js} +3 -3
- package/dist/ops/adamAdjust.js +1 -1
- package/dist/ops/adamMoments.js +1 -1
- package/dist/ops/add16.js +1 -1
- package/dist/ops/appendCache.js +3 -3
- package/dist/ops/attentionMask.js +1 -1
- package/dist/ops/concat16.js +2 -2
- package/dist/ops/cpu/adamAdjust.js +7 -7
- package/dist/ops/cpu/adamMoments.js +5 -5
- package/dist/ops/cpu/appendCache.js +6 -6
- package/dist/ops/cpu/attentionMask.js +6 -6
- package/dist/ops/cpu/fusedSoftmax.js +5 -5
- package/dist/ops/cpu/gatherSub.js +7 -7
- package/dist/ops/cpu/gelu.js +5 -5
- package/dist/ops/cpu/matMul16.js +2 -2
- package/dist/ops/cpu/matMulGelu.js +3 -3
- package/dist/ops/cpu/matMulMul.js +5 -5
- package/dist/ops/cpu/mulDropout.js +1 -1
- package/dist/ops/cpu/normRMS.js +5 -5
- package/dist/ops/cpu/qkv.js +3 -3
- package/dist/ops/cpu/rope.js +9 -9
- package/dist/ops/cpu/scatterSub.js +5 -5
- package/dist/ops/dot16.js +2 -2
- package/dist/ops/gatherSub.js +1 -1
- package/dist/ops/gelu.js +2 -2
- package/dist/ops/grads/add16.js +1 -1
- package/dist/ops/grads/attentionMask.js +2 -2
- package/dist/ops/grads/gelu.js +2 -2
- package/dist/ops/grads/matMul16.js +3 -3
- package/dist/ops/grads/matMulGelu.js +5 -5
- package/dist/ops/grads/normRMS.js +6 -6
- package/dist/ops/grads/pack16.js +3 -3
- package/dist/ops/grads/qkv.js +9 -9
- package/dist/ops/grads/rope.js +2 -2
- package/dist/ops/grads/softmax16.js +1 -1
- package/dist/ops/grads/unpack16.js +2 -2
- package/dist/ops/matMul16.js +3 -3
- package/dist/ops/matMulGelu.js +2 -2
- package/dist/ops/matMulMul.js +1 -1
- package/dist/ops/mul16.js +1 -1
- package/dist/ops/mulDrop.js +1 -1
- package/dist/ops/normRMS.js +1 -1
- package/dist/ops/pack16.js +2 -2
- package/dist/ops/qkv.js +1 -1
- package/dist/ops/reshape16.js +6 -6
- package/dist/ops/rope.js +2 -2
- package/dist/ops/scatterSub.js +1 -1
- package/dist/ops/slice16.js +2 -2
- package/dist/ops/softmax16.js +1 -1
- package/dist/ops/sub16.js +1 -1
- package/dist/ops/sum16.js +2 -2
- package/dist/ops/transpose16.js +6 -6
- package/dist/ops/unpack16.js +2 -2
- package/dist/ops/webgl/adamAdjust.js +2 -2
- package/dist/ops/webgl/adamMoments.js +1 -1
- package/dist/ops/webgl/appendCache.js +1 -1
- package/dist/ops/webgl/attentionMask.js +4 -4
- package/dist/ops/webgl/fusedSoftmax.js +6 -6
- package/dist/ops/webgl/gatherSub.js +1 -1
- package/dist/ops/webgl/gelu.js +2 -2
- package/dist/ops/webgl/log.js +3 -3
- package/dist/ops/webgl/matMul16.js +10 -10
- package/dist/ops/webgl/matMulGelu.js +4 -4
- package/dist/ops/webgl/matMulMul.js +2 -2
- package/dist/ops/webgl/mulDropout.js +1 -1
- package/dist/ops/webgl/normRMS.js +2 -2
- package/dist/ops/webgl/qkv.js +1 -1
- package/dist/ops/webgl/rope.js +4 -4
- package/dist/ops/webgl/scatterSub.js +1 -1
- package/dist/ops/webgpu/adamAdjust.js +3 -3
- package/dist/ops/webgpu/adamMoments.js +5 -5
- package/dist/ops/webgpu/add16.js +1 -1
- package/dist/ops/webgpu/appendCache.js +3 -3
- package/dist/ops/webgpu/attentionMask.js +5 -5
- package/dist/ops/webgpu/attentionMask32_program.js +2 -2
- package/dist/ops/webgpu/concat16.js +5 -5
- package/dist/ops/webgpu/gatherSub.js +3 -3
- package/dist/ops/webgpu/gelu.js +3 -3
- package/dist/ops/webgpu/matMul16.js +19 -19
- package/dist/ops/webgpu/matMul16_program.js +2 -2
- package/dist/ops/webgpu/mul16.js +1 -1
- package/dist/ops/webgpu/normRMS.js +2 -2
- package/dist/ops/webgpu/normRMSGrad.js +4 -4
- package/dist/ops/webgpu/pack16.js +3 -3
- package/dist/ops/webgpu/pack16_program.js +2 -2
- package/dist/ops/webgpu/qkv.js +4 -4
- package/dist/ops/webgpu/rope.js +3 -3
- package/dist/ops/webgpu/scatterSub.js +3 -3
- package/dist/ops/webgpu/slice16.js +4 -4
- package/dist/ops/webgpu/softmax16.js +4 -4
- package/dist/ops/webgpu/softmax16_program.js +2 -2
- package/dist/ops/webgpu/softmax16_subgroup_program.js +2 -2
- package/dist/ops/webgpu/softmax16grad.js +1 -1
- package/dist/ops/webgpu/sub16.js +1 -1
- package/dist/ops/webgpu/sum16.js +5 -5
- package/dist/ops/webgpu/transpose16.js +2 -2
- package/dist/ops/webgpu/transpose16_program.js +2 -2
- package/dist/ops/webgpu/transpose16_shared_program.js +3 -3
- package/dist/ops/webgpu/unpack16.js +5 -5
- package/dist/ops/webgpu/utils/binary_op.js +3 -3
- package/dist/ops/webgpu/utils/reductions.js +4 -4
- package/dist/{ops-FJapAPfm.js → ops-C2_OXuZ4.js} +35 -35
- package/dist/{pack16-k4jq6aMX.js → pack16-atD0eYRm.js} +6 -6
- package/dist/patches/webgpu_backend.js +8 -8
- package/dist/patches/webgpu_base.js +1 -1
- package/dist/patches/webgpu_program.js +2 -2
- package/dist/{random_width-UGQn4OWb.js → random_width-BN4wGJaW.js} +33 -33
- package/dist/{range-CuGvVN2c.js → range-DKmP1-OQ.js} +1 -1
- package/dist/relu-BsXmGzzu.js +9 -0
- package/dist/{reshape-CkjKPPqB.js → reshape-BI0yzp1T.js} +1 -1
- package/dist/{resize_nearest_neighbor-DB8k9KN_.js → resize_nearest_neighbor-BA_BX-ub.js} +25 -25
- package/dist/{rope-BmZmp9uP.js → rope-DJ7Y7c-u.js} +1 -1
- package/dist/{scatter_nd_util-BY22Cc-C.js → scatter_nd_util-k9MUVUkn.js} +1 -1
- package/dist/{selu_util-BuLbmbrl.js → selu_util-DyW0X1WG.js} +5 -5
- package/dist/{shared-B7USJZgw.js → shared-Q3BS6T03.js} +1 -1
- package/dist/{shared-BQboIImQ.js → shared-nnSWpC3u.js} +6 -6
- package/dist/{slice-Aqy7KbJh.js → slice-wBNvzVyz.js} +3 -3
- package/dist/{slice_util-D8CQRenR.js → slice_util-zN8KFC5I.js} +7 -7
- package/dist/{softmax-faLoUZVT.js → softmax-DfuYyjMh.js} +1 -1
- package/dist/split-BYrLboMq.js +9 -0
- package/dist/squeeze-Bk8Brcct.js +10 -0
- package/dist/{stack-WJK22CFn.js → stack-CDWShFHF.js} +1 -1
- package/dist/{step-dXR33iOg.js → step-BS5JXRR6.js} +14 -14
- package/dist/sum-BPUfDB2X.js +11 -0
- package/dist/{tensor-BQqrDvpx.js → tensor-CEt9Nm2s.js} +1 -1
- package/dist/{tensor1d-LxP9asMm.js → tensor1d-Cc_KCIDg.js} +1 -1
- package/dist/{tensor2d-BN1sSfQO.js → tensor2d-BN97fF71.js} +1 -1
- package/dist/{tensor4d-DVwr7pLF.js → tensor4d-vuDDgdUI.js} +1 -1
- package/dist/{tfjs_backend-Vi4JfLzT.js → tfjs_backend-806hyYve.js} +36 -36
- package/dist/tile-OWUvpIVt.js +11 -0
- package/dist/tokeniser/BaseTokeniser.d.ts +6 -8
- package/dist/tokeniser/BaseTokeniser.js +6 -6
- package/dist/tokeniser/CharTokeniser.d.ts +6 -6
- package/dist/tokeniser/CharTokeniser.js +26 -26
- package/dist/tokeniser/bpe.d.ts +6 -6
- package/dist/tokeniser/bpe.js +9 -9
- package/dist/tokeniser/type.d.ts +6 -8
- package/dist/training/Adam.js +2 -2
- package/dist/training/AdamExt.js +1 -1
- package/dist/training/DatasetBuilder.d.ts +1 -1
- package/dist/training/DatasetBuilder.js +29 -29
- package/dist/training/FullTrainer.js +1 -1
- package/dist/training/Trainer.d.ts +5 -4
- package/dist/training/Trainer.js +37 -40
- package/dist/training/sparseCrossEntropy.js +3 -3
- package/dist/training/tasks/ConversationTask.d.ts +11 -0
- package/dist/training/tasks/ConversationTask.js +26 -0
- package/dist/training/tasks/PretrainingTask.d.ts +11 -0
- package/dist/training/tasks/PretrainingTask.js +34 -0
- package/dist/training/tasks/StartSentenceTask.d.ts +12 -0
- package/dist/training/tasks/StartSentenceTask.js +42 -0
- package/dist/training/tasks/Task.d.ts +8 -0
- package/dist/training/tasks/Task.js +44 -0
- package/dist/{transpose-JawVKyZy.js → transpose-BUkQCJp9.js} +7 -7
- package/dist/{unsorted_segment_sum-LAbmE9G4.js → unsorted_segment_sum-BljxHhCY.js} +78 -78
- package/dist/utilities/dummy.js +3 -3
- package/dist/utilities/multinomialCPU.js +2 -2
- package/dist/utilities/packed.js +1 -1
- package/dist/utilities/performance.js +1 -1
- package/dist/utilities/profile.js +1 -1
- package/dist/utilities/safetensors.js +2 -2
- package/dist/utilities/sentences.d.ts +1 -1
- package/dist/utilities/sentences.js +11 -11
- package/dist/utilities/weights.js +2 -2
- package/dist/{variable-DQ9yYgEU.js → variable-DPt_Iuog.js} +1 -1
- package/dist/{webgpu_program-CAE4RICo.js → webgpu_program-BpWRlghH.js} +1 -1
- package/dist/{webgpu_util-BdovYhXr.js → webgpu_util-DMiKzzQM.js} +7 -7
- package/dist/{zeros-DeiE2zTa.js → zeros-5YROwwUH.js} +2 -2
- package/dist/{zeros_like-BAz3iKru.js → zeros_like-De4n1C3m.js} +57 -57
- package/package.json +1 -1
- package/dist/clip_by_value-Dn5tzexi.js +0 -12
- package/dist/concat-C6X3AAlQ.js +0 -17
- package/dist/expand_dims-BzfJK2uc.js +0 -11
- package/dist/gather-BcO5UQNJ.js +0 -9
- package/dist/mat_mul-DxpNTCRz.js +0 -11
- package/dist/relu-Cf80uA2p.js +0 -9
- package/dist/split-BNz5jcGc.js +0 -9
- package/dist/squeeze--YMgaAAf.js +0 -10
- package/dist/sum-BdplSvq_.js +0 -11
- package/dist/tile-CvN_LyVr.js +0 -11
package/dist/ops/appendCache.js
CHANGED
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
import { e as a } from "../index-
|
|
1
|
+
import { e as a } from "../index-Duu1Lvvv.js";
|
|
2
2
|
import "./cpu/appendCache.js";
|
|
3
3
|
import "./webgl/appendCache.js";
|
|
4
4
|
import { isPackedTensor as c } from "../utilities/packed.js";
|
|
5
|
-
import { c as t } from "../concat-
|
|
6
|
-
import { z as f } from "../zeros-
|
|
5
|
+
import { c as t } from "../concat-CSm2rMwe.js";
|
|
6
|
+
import { z as f } from "../zeros-5YROwwUH.js";
|
|
7
7
|
function C(r, o, n, p) {
|
|
8
8
|
if (!p) {
|
|
9
9
|
const e = r.shape[2], s = c(r);
|
package/dist/ops/concat16.js
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { isPackedTensor as o } from "../utilities/packed.js";
|
|
2
|
-
import { e } from "../index-
|
|
3
|
-
import { c } from "../concat-
|
|
2
|
+
import { e } from "../index-Duu1Lvvv.js";
|
|
3
|
+
import { c } from "../concat-CSm2rMwe.js";
|
|
4
4
|
function p(r, n) {
|
|
5
5
|
return o(r[0]) ? e().runKernel("Concat16", r, { axis: n ?? -1 }) : c(r, n);
|
|
6
6
|
}
|
|
@@ -1,18 +1,18 @@
|
|
|
1
|
-
import {
|
|
2
|
-
function
|
|
1
|
+
import { f as k, j as t, l as i, m as z, n as A } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
function C(c) {
|
|
3
3
|
const { moments: n, value: r } = c.inputs, { beta1: l, beta2: m, epsilon: u, learningRate: d } = c.attrs, e = n.shape.length, a = new Array(e).fill(0), s = n.shape.slice();
|
|
4
4
|
s[e - 1] = 1;
|
|
5
5
|
const o = a.slice();
|
|
6
6
|
o[e - 1] = 1;
|
|
7
|
-
const b = s.slice(), p = n.slice(a, s).squeeze([e - 1]), M = n.slice(o, b).squeeze([e - 1]),
|
|
7
|
+
const b = s.slice(), p = n.slice(a, s).squeeze([e - 1]), M = n.slice(o, b).squeeze([e - 1]), f = t(p, l), g = t(M, m);
|
|
8
8
|
return i(
|
|
9
|
-
|
|
9
|
+
z(t(f, i(A(g), u ?? 1e-8)), -d),
|
|
10
10
|
r
|
|
11
11
|
);
|
|
12
12
|
}
|
|
13
|
-
const
|
|
13
|
+
const h = {
|
|
14
14
|
kernelName: "AdamAdjust",
|
|
15
15
|
backendName: "cpu",
|
|
16
|
-
kernelFunc:
|
|
16
|
+
kernelFunc: C
|
|
17
17
|
};
|
|
18
|
-
h
|
|
18
|
+
k(h);
|
|
@@ -1,16 +1,16 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { s as
|
|
3
|
-
function
|
|
1
|
+
import { f as p } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { s as f } from "../../stack-CDWShFHF.js";
|
|
3
|
+
function b(t) {
|
|
4
4
|
const { moments: n, gradient: c } = t.inputs, { beta1: o, beta2: m } = t.attrs, e = n.shape.length, a = new Array(e).fill(0), s = n.shape.slice();
|
|
5
5
|
s[e - 1] = 1;
|
|
6
6
|
const i = a.slice();
|
|
7
7
|
i[e - 1] = 1;
|
|
8
8
|
const r = s.slice(), l = n.slice(a, s).squeeze([e - 1]), u = n.slice(i, r).squeeze([e - 1]), M = l.mul(o).add(c.mul(1 - o)), d = u.mul(m).add(c.square().mul(1 - m));
|
|
9
|
-
return
|
|
9
|
+
return f([M, d], -1);
|
|
10
10
|
}
|
|
11
11
|
const g = {
|
|
12
12
|
kernelName: "AdamMoments",
|
|
13
13
|
backendName: "cpu",
|
|
14
|
-
kernelFunc:
|
|
14
|
+
kernelFunc: b
|
|
15
15
|
};
|
|
16
16
|
p(g);
|
|
@@ -1,13 +1,13 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { c as
|
|
1
|
+
import { f as d } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { c as h } from "../../concat-CSm2rMwe.js";
|
|
3
3
|
function u(p) {
|
|
4
4
|
const { cache: n, item: s } = p.inputs, { maxSize: i, pastLen: c } = p.attrs, t = n.shape[0], o = n.shape[1], a = n.shape[3], e = s.shape[2];
|
|
5
5
|
if (c + e <= i) {
|
|
6
|
-
const
|
|
7
|
-
return
|
|
6
|
+
const l = n.slice([0, 0, 0, 0], [t, o, c, a]), m = n.slice([0, 0, c + e, 0], [t, o, i - c - e, a]), r = e < e ? s.slice([0, 0, 0, 0], [t, o, e, a]) : s, k = h([l, r, m], 2);
|
|
7
|
+
return l.dispose(), m.dispose(), r !== s && r.dispose(), k;
|
|
8
8
|
}
|
|
9
|
-
const
|
|
10
|
-
return
|
|
9
|
+
const f = n.slice([0, 0, e, 0], [t, o, i - e, a]), C = h([f, s], 2);
|
|
10
|
+
return f.dispose(), C;
|
|
11
11
|
}
|
|
12
12
|
const w = {
|
|
13
13
|
kernelName: "AppendCache",
|
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { l as N } from "../../ops-
|
|
3
|
-
import { o as b } from "../../ones-
|
|
4
|
-
import { z as A } from "../../zeros-
|
|
5
|
-
import { w as I } from "../../resize_nearest_neighbor-
|
|
6
|
-
import { m as g } from "../../mat_mul-
|
|
1
|
+
import { f as o, h as d, b as u } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { l as N } from "../../ops-C2_OXuZ4.js";
|
|
3
|
+
import { o as b } from "../../ones-CBI1AQjb.js";
|
|
4
|
+
import { z as A } from "../../zeros-5YROwwUH.js";
|
|
5
|
+
import { w as I } from "../../resize_nearest_neighbor-BA_BX-ub.js";
|
|
6
|
+
import { m as g } from "../../mat_mul-Bn2BDpT4.js";
|
|
7
7
|
function a(n) {
|
|
8
8
|
const { q: s, k: e } = n.inputs, { divisor: r } = n.attrs, c = s.shape[2], t = e.shape[2], m = N.bandPart(b([t, t]), -1, 0).cast("bool"), i = A([t, t]), l = d([t, t], Number.NEGATIVE_INFINITY), f = I(m, i, l), k = g(s, e, !1, !0).mul(u(r)), p = f.slice([0, 0], [c, t]).expandDims(0).expandDims(0);
|
|
9
9
|
return k.add(p);
|
|
@@ -1,17 +1,17 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { s as
|
|
1
|
+
import { f as e } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { s as f } from "../../softmax-DfuYyjMh.js";
|
|
3
3
|
function n(t) {
|
|
4
4
|
const { inputs: s, attrs: a } = t, { logits: o } = s, { dim: i, dropoutRate: r } = a;
|
|
5
5
|
if (!o)
|
|
6
6
|
throw new Error("Error in softmax: input logits is null");
|
|
7
|
-
return r !== void 0 && r > 0 && console.warn("Dropout in fusedSoftmax not implemented for CPU backend, skipping dropout."),
|
|
7
|
+
return r !== void 0 && r > 0 && console.warn("Dropout in fusedSoftmax not implemented for CPU backend, skipping dropout."), f(o, i);
|
|
8
8
|
}
|
|
9
|
-
const
|
|
9
|
+
const m = {
|
|
10
10
|
kernelName: "FusedSoftmax",
|
|
11
11
|
backendName: "cpu",
|
|
12
12
|
kernelFunc: n
|
|
13
13
|
};
|
|
14
|
-
e(
|
|
14
|
+
e(m);
|
|
15
15
|
const u = {
|
|
16
16
|
kernelName: "FusedSoftmax",
|
|
17
17
|
backendName: "tensorflow",
|
|
@@ -1,14 +1,14 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { r as
|
|
3
|
-
import { s as
|
|
4
|
-
function
|
|
1
|
+
import { o as u, q as c, E as g, aj as h, f as m, c as p } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { r as f } from "../../range-DKmP1-OQ.js";
|
|
3
|
+
import { s as l } from "../../stack-CDWShFHF.js";
|
|
4
|
+
function N(e, t) {
|
|
5
5
|
const n = c(t, "indices", "gatherND", "int32"), s = { params: c(e, "x", "gatherND", "string_or_numeric"), indices: n };
|
|
6
6
|
return g.runKernel(h, s);
|
|
7
7
|
}
|
|
8
|
-
const b = /* @__PURE__ */ u({ gatherND_:
|
|
8
|
+
const b = /* @__PURE__ */ u({ gatherND_: N });
|
|
9
9
|
function d(e) {
|
|
10
|
-
const { values: t, labels: n, logits: r } = e.inputs, s = n.shape[0], a =
|
|
11
|
-
return p(t,
|
|
10
|
+
const { values: t, labels: n, logits: r } = e.inputs, s = n.shape[0], a = f(0, s, 1, "int32"), o = l([a, n], 1), i = b(r, o);
|
|
11
|
+
return p(t, i);
|
|
12
12
|
}
|
|
13
13
|
const k = {
|
|
14
14
|
kernelName: "EfficientGatherSub",
|
package/dist/ops/cpu/gelu.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { f as t, t as d } from "../../index-Duu1Lvvv.js";
|
|
2
2
|
const o = 0.7978845608028654, c = 0.044715;
|
|
3
3
|
function m(r) {
|
|
4
4
|
const { inputs: u } = r, { x: n } = u, e = n;
|
|
@@ -26,15 +26,15 @@ function i(r) {
|
|
|
26
26
|
return u.mul(N);
|
|
27
27
|
});
|
|
28
28
|
}
|
|
29
|
-
const
|
|
29
|
+
const x = {
|
|
30
30
|
kernelName: "GeluGrad",
|
|
31
31
|
backendName: "cpu",
|
|
32
32
|
kernelFunc: i
|
|
33
33
|
};
|
|
34
|
-
t(
|
|
35
|
-
const
|
|
34
|
+
t(x);
|
|
35
|
+
const h = {
|
|
36
36
|
kernelName: "GeluGrad",
|
|
37
37
|
backendName: "tensorflow",
|
|
38
38
|
kernelFunc: i
|
|
39
39
|
};
|
|
40
|
-
t(
|
|
40
|
+
t(h);
|
package/dist/ops/cpu/matMul16.js
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { isPackedTensor as t } from "../../utilities/packed.js";
|
|
2
|
-
import {
|
|
3
|
-
import { m as l } from "../../mat_mul-
|
|
2
|
+
import { f as p } from "../../index-Duu1Lvvv.js";
|
|
3
|
+
import { m as l } from "../../mat_mul-Bn2BDpT4.js";
|
|
4
4
|
function m(e) {
|
|
5
5
|
const { A: n, B: r } = e.inputs, { transposeA: o, transposeB: s } = e.attrs, a = !t(n), c = !t(r);
|
|
6
6
|
if (a && c)
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { g as M, d as i } from "../../gelu-
|
|
3
|
-
import { m as k } from "../../mat_mul-
|
|
1
|
+
import { f as e, t as m } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { g as M, d as i } from "../../gelu-BK1k-n1i.js";
|
|
3
|
+
import { m as k } from "../../mat_mul-Bn2BDpT4.js";
|
|
4
4
|
function c(t) {
|
|
5
5
|
const { inputs: u } = t, { x: n, kernel: r } = u, a = n, l = r;
|
|
6
6
|
return m(() => {
|
|
@@ -1,20 +1,20 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { f as e, t as i } from "../../index-Duu1Lvvv.js";
|
|
2
2
|
function n(t) {
|
|
3
3
|
const { inputs: r, attrs: l } = t, { transposeA: o, transposeB: s } = l, { x: u, kernel: a, y: c } = r, m = u, k = a, M = c;
|
|
4
4
|
return i(() => m.matMul(k, o, s).mul(M));
|
|
5
5
|
}
|
|
6
|
-
const
|
|
6
|
+
const f = {
|
|
7
7
|
kernelName: "MatMulMul",
|
|
8
8
|
backendName: "cpu",
|
|
9
9
|
kernelFunc: n
|
|
10
10
|
};
|
|
11
|
-
e(
|
|
12
|
-
const
|
|
11
|
+
e(f);
|
|
12
|
+
const p = {
|
|
13
13
|
kernelName: "MatMulMul",
|
|
14
14
|
backendName: "tensorflow",
|
|
15
15
|
kernelFunc: n
|
|
16
16
|
};
|
|
17
|
-
e(
|
|
17
|
+
e(p);
|
|
18
18
|
const g = {
|
|
19
19
|
kernelName: "MatMulMul",
|
|
20
20
|
backendName: "webgpu",
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { f as e, m as t } from "../../index-Duu1Lvvv.js";
|
|
2
2
|
function n(o) {
|
|
3
3
|
const { inputs: r } = o, { a: l, b: u } = r;
|
|
4
4
|
return console.warn("Using fallback mulDrop implementation without dropout."), t(l, u);
|
package/dist/ops/cpu/normRMS.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { f as o, t as d } from "../../index-Duu1Lvvv.js";
|
|
2
2
|
function i(t) {
|
|
3
3
|
const { inputs: e } = t, { x: n, gamma: s } = e, r = n, a = s;
|
|
4
4
|
return d(() => {
|
|
@@ -6,12 +6,12 @@ function i(t) {
|
|
|
6
6
|
return r.mul(u).mul(a);
|
|
7
7
|
});
|
|
8
8
|
}
|
|
9
|
-
const
|
|
9
|
+
const k = {
|
|
10
10
|
kernelName: "RMSNorm",
|
|
11
11
|
backendName: "cpu",
|
|
12
12
|
kernelFunc: i
|
|
13
13
|
};
|
|
14
|
-
o(
|
|
14
|
+
o(k);
|
|
15
15
|
const g = {
|
|
16
16
|
kernelName: "RMSNorm",
|
|
17
17
|
backendName: "tensorflow",
|
|
@@ -21,8 +21,8 @@ o(g);
|
|
|
21
21
|
function N(t) {
|
|
22
22
|
const { dy: e, x: n, gamma: s } = t.inputs;
|
|
23
23
|
return d(() => {
|
|
24
|
-
const r = n.shape[n.shape.length - 1], a = n.square().mean(-1, !0), m = a.add(1e-8).rsqrt(), u = n.mul(m), l = e.mul(u).sum([0, 1]), c = e.mul(s),
|
|
25
|
-
return [c.mul(m).sub(n.mul(
|
|
24
|
+
const r = n.shape[n.shape.length - 1], a = n.square().mean(-1, !0), m = a.add(1e-8).rsqrt(), u = n.mul(m), l = e.mul(u).sum([0, 1]), c = e.mul(s), f = c.mul(n).sum(-1, !0).div(r);
|
|
25
|
+
return [c.mul(m).sub(n.mul(f).mul(m).div(a.add(1e-8))), l];
|
|
26
26
|
});
|
|
27
27
|
}
|
|
28
28
|
const S = {
|
package/dist/ops/cpu/qkv.js
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { r as o } from "../../reshape-
|
|
3
|
-
import { s as x } from "../../split-
|
|
1
|
+
import { f as q } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { r as o } from "../../reshape-BI0yzp1T.js";
|
|
3
|
+
import { s as x } from "../../split-BYrLboMq.js";
|
|
4
4
|
function v(p) {
|
|
5
5
|
const { x: c, kernel: K } = p.inputs, { heads: n, packed: C } = p.attrs;
|
|
6
6
|
if (C)
|
package/dist/ops/cpu/rope.js
CHANGED
|
@@ -1,20 +1,20 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { r as y } from "../../range-
|
|
3
|
-
import { g as F } from "../../gather-
|
|
4
|
-
import { s as E } from "../../stack-
|
|
5
|
-
import { c as T } from "../../concat-
|
|
1
|
+
import { f as I } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { r as y } from "../../range-DKmP1-OQ.js";
|
|
3
|
+
import { g as F } from "../../gather-BSULDalH.js";
|
|
4
|
+
import { s as E } from "../../stack-CDWShFHF.js";
|
|
5
|
+
import { c as T } from "../../concat-CSm2rMwe.js";
|
|
6
6
|
function U(c, r, p, e, n) {
|
|
7
7
|
const t = e.shape[3], s = p;
|
|
8
8
|
if (s > t) return e;
|
|
9
|
-
const o = e.shape[2], i = s / 2, a = r.slice([n, 0, 0], [o, i, 1]).reshape([1, 1, o, i]), d = c.slice([n, 0, 0], [o, i, 1]).reshape([1, 1, o, i]), l = e.shape[0], m = e.shape[1],
|
|
10
|
-
const C = k.slice([0, 0, 0, 0], [l, m, o, s]), R = s < t ? k.slice([0, 0, 0, s], [l, m, o, t - s]) : null, u = F(C,
|
|
11
|
-
u.dispose(),
|
|
9
|
+
const o = e.shape[2], i = s / 2, a = r.slice([n, 0, 0], [o, i, 1]).reshape([1, 1, o, i]), d = c.slice([n, 0, 0], [o, i, 1]).reshape([1, 1, o, i]), l = e.shape[0], m = e.shape[1], h = y(0, s, 2, "int32"), g = y(1, s, 2, "int32"), D = ((k) => {
|
|
10
|
+
const C = k.slice([0, 0, 0, 0], [l, m, o, s]), R = s < t ? k.slice([0, 0, 0, s], [l, m, o, t - s]) : null, u = F(C, h, 3), f = F(C, g, 3), v = u.mul(a), N = f.mul(d), S = v.sub(N), P = f.mul(a), b = u.mul(d), x = P.add(b);
|
|
11
|
+
u.dispose(), f.dispose(), a.dispose(), d.dispose(), v.dispose(), N.dispose(), P.dispose(), b.dispose();
|
|
12
12
|
const K = E([S, x], -1);
|
|
13
13
|
S.dispose(), x.dispose();
|
|
14
14
|
const w = K.reshape([l, m, o, s]);
|
|
15
15
|
return K.dispose(), R ? T([w, R], 3) : w;
|
|
16
16
|
})(e);
|
|
17
|
-
return
|
|
17
|
+
return h.dispose(), g.dispose(), D;
|
|
18
18
|
}
|
|
19
19
|
function B(c) {
|
|
20
20
|
const { x: r } = c.inputs, { pastLen: p, negSin: e, ropeCache: n } = c.attrs, t = r.shape[3], s = e ? n.getNegSin() : n.getSin(), o = n.getCos();
|
|
@@ -1,8 +1,8 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { v as D } from "../../scatter_nd_util-
|
|
3
|
-
import { r as k } from "../../range-
|
|
4
|
-
import { s as v } from "../../stack-
|
|
5
|
-
import { o as E } from "../../ones-
|
|
1
|
+
import { o as f, u as g, q as r, E as l, ai as N, f as b, c as S, m as h } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { v as D } from "../../scatter_nd_util-k9MUVUkn.js";
|
|
3
|
+
import { r as k } from "../../range-DKmP1-OQ.js";
|
|
4
|
+
import { s as v } from "../../stack-CDWShFHF.js";
|
|
5
|
+
import { o as E } from "../../ones-CBI1AQjb.js";
|
|
6
6
|
function I(a, e, s) {
|
|
7
7
|
g(s);
|
|
8
8
|
const n = r(a, "indices", "scatterND", "int32"), t = r(e, "updates", "scatterND");
|
package/dist/ops/dot16.js
CHANGED
|
@@ -1,8 +1,8 @@
|
|
|
1
|
-
import { b as d } from "../matMul16-
|
|
1
|
+
import { b as d } from "../matMul16-xswmhSuF.js";
|
|
2
2
|
import { transpose16 as w } from "./transpose16.js";
|
|
3
3
|
import { reshape16 as n } from "./reshape16.js";
|
|
4
4
|
import { isPackedTensor as p } from "../utilities/packed.js";
|
|
5
|
-
import { d as x } from "../tfjs_backend-
|
|
5
|
+
import { d as x } from "../tfjs_backend-806hyYve.js";
|
|
6
6
|
function E(e, s, h = !1, c = !1) {
|
|
7
7
|
if (!p(e) && !p(s))
|
|
8
8
|
return x(e, s);
|
package/dist/ops/gatherSub.js
CHANGED
package/dist/ops/gelu.js
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
import "../index-
|
|
1
|
+
import "../index-Duu1Lvvv.js";
|
|
2
2
|
import "./cpu/gelu.js";
|
|
3
3
|
import "./webgl/gelu.js";
|
|
4
|
-
import { d as e, g as i } from "../gelu-
|
|
4
|
+
import { d as e, g as i } from "../gelu-BK1k-n1i.js";
|
|
5
5
|
export {
|
|
6
6
|
e as dGelu,
|
|
7
7
|
i as gelu
|
package/dist/ops/grads/add16.js
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { m as o } from "../../matMul16-
|
|
1
|
+
import { i as m } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { m as o } from "../../matMul16-xswmhSuF.js";
|
|
3
3
|
import { transpose16 as c } from "../transpose16.js";
|
|
4
4
|
const l = {
|
|
5
5
|
kernelName: "AttentionMask",
|
package/dist/ops/grads/gelu.js
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import "../../index-
|
|
2
|
-
import { a } from "../../matMul16-
|
|
3
|
-
import "../../gelu-
|
|
1
|
+
import "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { a } from "../../matMul16-xswmhSuF.js";
|
|
3
|
+
import "../../gelu-BK1k-n1i.js";
|
|
4
4
|
import "../transpose16.js";
|
|
5
5
|
import "../reshape16.js";
|
|
6
6
|
export {
|
|
@@ -1,17 +1,17 @@
|
|
|
1
|
-
import {
|
|
2
|
-
function
|
|
1
|
+
import { i as a, e as o } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
function i(e, n, r) {
|
|
3
3
|
return o().runKernel("MatMulGeluGrad", { dy: e, x: n, kernel: r });
|
|
4
4
|
}
|
|
5
|
-
const
|
|
5
|
+
const s = {
|
|
6
6
|
kernelName: "MatMulGelu",
|
|
7
7
|
inputsToSave: ["x", "kernel"],
|
|
8
8
|
outputsToSave: [],
|
|
9
9
|
gradFunc: (e, n) => {
|
|
10
|
-
const [r, t] = n, [u, l] =
|
|
10
|
+
const [r, t] = n, [u, l] = i(e, r, t);
|
|
11
11
|
return {
|
|
12
12
|
x: () => u,
|
|
13
13
|
kernel: () => l
|
|
14
14
|
};
|
|
15
15
|
}
|
|
16
16
|
};
|
|
17
|
-
a(
|
|
17
|
+
a(s);
|
|
@@ -1,20 +1,20 @@
|
|
|
1
|
-
import {
|
|
2
|
-
function
|
|
1
|
+
import { i as t, e as g } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
function i(r, a, n) {
|
|
3
3
|
return g().runKernel("RMSNormGrad", { dy: r, x: a, gamma: n });
|
|
4
4
|
}
|
|
5
|
-
const
|
|
5
|
+
const s = {
|
|
6
6
|
kernelName: "RMSNorm",
|
|
7
7
|
inputsToSave: ["x", "gamma"],
|
|
8
8
|
outputsToSave: [],
|
|
9
9
|
gradFunc: (r, a) => {
|
|
10
|
-
const [n, e] = a, [m, o] =
|
|
10
|
+
const [n, e] = a, [m, o] = i(r, n, e);
|
|
11
11
|
return {
|
|
12
12
|
x: () => m,
|
|
13
13
|
gamma: () => o
|
|
14
14
|
};
|
|
15
15
|
}
|
|
16
16
|
};
|
|
17
|
-
t(
|
|
17
|
+
t(s);
|
|
18
18
|
export {
|
|
19
|
-
|
|
19
|
+
s as normRMSGradConfig
|
|
20
20
|
};
|
package/dist/ops/grads/pack16.js
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import "../../index-
|
|
2
|
-
import { b as i } from "../../pack16-
|
|
3
|
-
import "../../slice-
|
|
1
|
+
import "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { b as i } from "../../pack16-atD0eYRm.js";
|
|
3
|
+
import "../../slice-wBNvzVyz.js";
|
|
4
4
|
export {
|
|
5
5
|
i as packGradConfig
|
|
6
6
|
};
|
package/dist/ops/grads/qkv.js
CHANGED
|
@@ -1,34 +1,34 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { a as f } from "../../matMul16-
|
|
1
|
+
import { i as u } from "../../index-Duu1Lvvv.js";
|
|
2
|
+
import { a as f } from "../../matMul16-xswmhSuF.js";
|
|
3
3
|
import { concat16 as g } from "../concat16.js";
|
|
4
4
|
import { sum16 as l } from "../sum16.js";
|
|
5
|
-
import { s as k } from "../../squeeze
|
|
6
|
-
const
|
|
5
|
+
import { s as k } from "../../squeeze-Bk8Brcct.js";
|
|
6
|
+
const d = {
|
|
7
7
|
kernelName: "QKV",
|
|
8
8
|
inputsToSave: ["x", "kernel"],
|
|
9
9
|
outputsToSave: [],
|
|
10
10
|
gradFunc: (e, s) => {
|
|
11
11
|
const [r, n, t] = e, [a] = s, p = g([r, n, t], 1);
|
|
12
12
|
r.dispose(), n.dispose(), t.dispose();
|
|
13
|
-
const m = [a.shape[0], a.shape[1], 3 * a.shape[2]],
|
|
13
|
+
const m = [a.shape[0], a.shape[1], 3 * a.shape[2]], i = f.gradFunc(p, s, {
|
|
14
14
|
transposeA: !1,
|
|
15
15
|
transposeB: !1,
|
|
16
16
|
originalShape: m,
|
|
17
17
|
perm: [0, 2, 1, 3]
|
|
18
18
|
});
|
|
19
19
|
return p.dispose(), {
|
|
20
|
-
x: () =>
|
|
20
|
+
x: () => i.A(),
|
|
21
21
|
kernel: () => {
|
|
22
|
-
const o =
|
|
22
|
+
const o = i.B(), c = o.shape[0] === 1 ? k(o, [0]) : l(o, 0);
|
|
23
23
|
return o.dispose(), c;
|
|
24
24
|
}
|
|
25
25
|
};
|
|
26
26
|
}
|
|
27
27
|
};
|
|
28
28
|
function B(e, s, r) {
|
|
29
|
-
return
|
|
29
|
+
return d.gradFunc(e, [s, r], {});
|
|
30
30
|
}
|
|
31
|
-
u(
|
|
31
|
+
u(d);
|
|
32
32
|
export {
|
|
33
33
|
B as qkvGrad
|
|
34
34
|
};
|
package/dist/ops/grads/rope.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { i as n, e as a } from "../../index-Duu1Lvvv.js";
|
|
2
2
|
import { isPackedTensor as t } from "../../utilities/packed.js";
|
|
3
3
|
function s(r, e) {
|
|
4
4
|
return a().runKernel("Softmax16Grad", { dy: r, softmaxOutput: e });
|
package/dist/ops/matMul16.js
CHANGED
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
import "../index-
|
|
2
|
-
import { b as p, c as u, d as i, e as s, m as M } from "../matMul16-
|
|
1
|
+
import "../index-Duu1Lvvv.js";
|
|
2
|
+
import { b as p, c as u, d as i, e as s, m as M } from "../matMul16-xswmhSuF.js";
|
|
3
3
|
import "./webgl/matMul16.js";
|
|
4
4
|
import "./cpu/matMul16.js";
|
|
5
5
|
import "../utilities/packed.js";
|
|
6
|
-
import "../pack16-
|
|
6
|
+
import "../pack16-atD0eYRm.js";
|
|
7
7
|
export {
|
|
8
8
|
p as matMul16,
|
|
9
9
|
u as matMul16Gelu,
|
package/dist/ops/matMulGelu.js
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { e as u } from "../index-
|
|
1
|
+
import { e as u } from "../index-Duu1Lvvv.js";
|
|
2
2
|
import "./cpu/matMulGelu.js";
|
|
3
|
-
import "../matMulGelu-
|
|
3
|
+
import "../matMulGelu-BpvgnYG8.js";
|
|
4
4
|
import "./grads/matMulGelu.js";
|
|
5
5
|
function M(r, e) {
|
|
6
6
|
return u().runKernel("MatMulGelu", { x: r, kernel: e });
|
package/dist/ops/matMulMul.js
CHANGED
package/dist/ops/mul16.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { m as t, e as u } from "../index-
|
|
1
|
+
import { m as t, e as u } from "../index-Duu1Lvvv.js";
|
|
2
2
|
import { isPackedTensor as n } from "../utilities/packed.js";
|
|
3
3
|
function i(r, e) {
|
|
4
4
|
return !n(r) && !n(e) ? t(r, e) : u().runKernel("Mul16", { a: r, b: e });
|
package/dist/ops/mulDrop.js
CHANGED
package/dist/ops/normRMS.js
CHANGED
package/dist/ops/pack16.js
CHANGED
package/dist/ops/qkv.js
CHANGED