@genai-fi/nanogpt 0.19.1 → 0.20.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/BaseTokeniser-DSg9zcYq.js +221 -0
- package/dist/DatasetBuilder-DgURD85T.js +712 -0
- package/dist/Generator.js +2 -11941
- package/dist/RealDiv-DBu0FQqT.js +362 -0
- package/dist/Reshape-CABOPB9d.js +94 -0
- package/dist/Reshape-DqO3r8BC.js +17 -0
- package/dist/TeachableLLM.d.ts +5 -5
- package/dist/TeachableLLM.js +2 -273
- package/dist/Trainer.js +2 -244
- package/dist/backend.js +12 -12
- package/dist/backend_util-Cg-roD1p.js +399 -0
- package/dist/binary_op_util-CrYk9LXL.js +103 -0
- package/dist/checks/appendCache.js +54 -21
- package/dist/checks/attentionMask.js +55 -36
- package/dist/checks/check.js +31 -19
- package/dist/checks/gelu.js +45 -17
- package/dist/checks/index.js +25 -25
- package/dist/checks/matMulGelu.js +83 -27
- package/dist/checks/normRMS.js +27 -15
- package/dist/checks/normRMSGrad.js +21 -11
- package/dist/checks/packUnpack.js +45 -17
- package/dist/checks/qkv.js +33 -33
- package/dist/checks/rope.js +29 -35
- package/dist/checks/weights.d.ts +1 -1
- package/dist/checks/weights.js +25 -29
- package/dist/chunk-BPntVaq0.js +23 -0
- package/dist/complex_util-CkazZsaH.js +60 -0
- package/dist/concat_util-CWDZCBlA.js +19 -0
- package/dist/data/docx.js +3044 -13
- package/dist/data/pdf.js +16 -13
- package/dist/data/textLoader.js +607 -112
- package/dist/dist-BewPQWjc.js +7572 -0
- package/dist/dist-DVmq73nz.js +8775 -0
- package/dist/dist-DXwIvKxl.js +896 -0
- package/dist/dist-VEU5mfO0.js +7545 -0
- package/dist/gelu-Bf1HW1RY.js +27 -0
- package/dist/gpgpu_math-DvLcCH6u.js +1612 -0
- package/dist/inference/types.js +0 -1
- package/dist/kernel_funcs_utils-HiXOOx3f.js +229 -0
- package/dist/layers/BaseLayer.d.ts +2 -2
- package/dist/layers/BaseLayer.js +75 -73
- package/dist/layers/CausalSelfAttention.d.ts +1 -1
- package/dist/layers/CausalSelfAttention.js +98 -85
- package/dist/layers/LoRA.js +47 -57
- package/dist/layers/MLP.d.ts +1 -1
- package/dist/layers/MLP.js +33 -43
- package/dist/layers/PositionEmbedding.d.ts +1 -1
- package/dist/layers/PositionEmbedding.js +26 -30
- package/dist/layers/RMSNorm.d.ts +1 -1
- package/dist/layers/RMSNorm.js +19 -21
- package/dist/layers/RoPECache.js +336 -49
- package/dist/layers/TiedEmbedding.d.ts +1 -1
- package/dist/layers/TiedEmbedding.js +30 -34
- package/dist/layers/TransformerBlock.d.ts +1 -1
- package/dist/layers/TransformerBlock.js +50 -39
- package/dist/layers/WeightStore.js +68 -75
- package/dist/loader/load.js +2 -68
- package/dist/loader/loadHF.d.ts +2 -2
- package/dist/loader/loadHF.js +2 -22
- package/dist/loader/loadTransformers.d.ts +1 -1
- package/dist/loader/loadTransformers.js +2 -44
- package/dist/loader/loadZipMeta.js +15 -15
- package/dist/loader/newZipLoad.js +2 -31
- package/dist/loader/oldZipLoad.d.ts +2 -2
- package/dist/loader/oldZipLoad.js +2 -80
- package/dist/loader/save.d.ts +5 -5
- package/dist/loader/save.js +2 -90
- package/dist/loader/types.d.ts +9 -8
- package/dist/loader/types.js +0 -1
- package/dist/main-CPjeMv0G.js +13500 -0
- package/dist/main.d.ts +1 -1
- package/dist/main.js +16 -109
- package/dist/matMul16-BNfZSnNM.js +81 -0
- package/dist/matMulGelu-CPTntosE.js +162 -0
- package/dist/models/NanoGPTV1.js +2 -99
- package/dist/models/NanoGPTV2.js +2 -90
- package/dist/models/config.js +34 -47
- package/dist/models/factory.d.ts +1 -1
- package/dist/models/factory.js +2 -16
- package/dist/models/model.d.ts +2 -2
- package/dist/models/model.js +2 -134
- package/dist/ops/adamAdjust.js +15 -6
- package/dist/ops/adamMoments.js +13 -6
- package/dist/ops/add16.js +9 -6
- package/dist/ops/appendCache.js +22 -19
- package/dist/ops/attentionMask.js +12 -6
- package/dist/ops/concat16.js +7 -8
- package/dist/ops/cpu/adamAdjust.js +15 -17
- package/dist/ops/cpu/adamMoments.js +15 -15
- package/dist/ops/cpu/appendCache.js +64 -22
- package/dist/ops/cpu/attentionMask.js +15 -21
- package/dist/ops/cpu/fusedSoftmax.js +21 -28
- package/dist/ops/cpu/gatherSub.js +11 -17
- package/dist/ops/cpu/gelu.js +34 -38
- package/dist/ops/cpu/matMul16.js +13 -14
- package/dist/ops/cpu/matMulGelu.js +39 -51
- package/dist/ops/cpu/matMulMul.js +19 -22
- package/dist/ops/cpu/mulDropout.js +19 -22
- package/dist/ops/cpu/normRMS.js +33 -37
- package/dist/ops/cpu/qkv.js +72 -40
- package/dist/ops/cpu/rope.js +79 -36
- package/dist/ops/cpu/scatterSub.js +11 -22
- package/dist/ops/dot16.js +28 -41
- package/dist/ops/dropout.js +10 -13
- package/dist/ops/dropout16.js +20 -23
- package/dist/ops/gatherSub.js +10 -6
- package/dist/ops/gelu.js +2 -8
- package/dist/ops/globalNorm.js +18 -12
- package/dist/ops/grads/add16.js +27 -26
- package/dist/ops/grads/attentionMask.js +26 -21
- package/dist/ops/grads/dropout16.js +0 -1
- package/dist/ops/grads/gelu.js +2 -5
- package/dist/ops/grads/matMul16.js +2 -9
- package/dist/ops/grads/matMulGelu.js +21 -16
- package/dist/ops/grads/mul16.js +0 -3
- package/dist/ops/grads/normRMS.js +34 -30
- package/dist/ops/grads/pack16.js +2 -6
- package/dist/ops/grads/qkv.js +44 -32
- package/dist/ops/grads/rope.js +2 -5
- package/dist/ops/grads/softmax16.js +21 -23
- package/dist/ops/grads/unpack16.js +2 -5
- package/dist/ops/grads/utils.js +9 -11
- package/dist/ops/matMul16.js +2 -13
- package/dist/ops/matMulGelu.js +16 -10
- package/dist/ops/matMulMul.js +13 -6
- package/dist/ops/mul16.js +42 -38
- package/dist/ops/mulDrop.js +12 -6
- package/dist/ops/normRMS.js +18 -15
- package/dist/ops/pack16.js +2 -5
- package/dist/ops/qkv.js +12 -6
- package/dist/ops/reshape16.js +31 -39
- package/dist/ops/rope.d.ts +1 -1
- package/dist/ops/rope.js +2 -7
- package/dist/ops/scatterSub.js +10 -6
- package/dist/ops/slice16.js +9 -7
- package/dist/ops/softmax16.js +7 -7
- package/dist/ops/sub16.js +9 -6
- package/dist/ops/sum16.js +12 -12
- package/dist/ops/transpose16.js +29 -37
- package/dist/ops/unpack16.js +2 -6
- package/dist/ops/webgl/adamAdjust.js +62 -29
- package/dist/ops/webgl/adamMoments.js +30 -26
- package/dist/ops/webgl/appendCache.js +30 -21
- package/dist/ops/webgl/attentionMask.js +43 -24
- package/dist/ops/webgl/dropout16.js +11 -10
- package/dist/ops/webgl/fusedSoftmax.js +69 -79
- package/dist/ops/webgl/gatherSub.js +27 -26
- package/dist/ops/webgl/gelu.js +32 -34
- package/dist/ops/webgl/log.js +14 -23
- package/dist/ops/webgl/matMul16.js +36 -44
- package/dist/ops/webgl/matMulGelu.js +2 -9
- package/dist/ops/webgl/matMulMul.js +23 -27
- package/dist/ops/webgl/mulDropout.js +31 -40
- package/dist/ops/webgl/normRMS.js +92 -71
- package/dist/ops/webgl/qkv.js +35 -27
- package/dist/ops/webgl/rope.js +37 -21
- package/dist/ops/webgl/scatterSub.js +27 -26
- package/dist/ops/webgpu/adamAdjust.js +59 -39
- package/dist/ops/webgpu/adamMoments.js +62 -46
- package/dist/ops/webgpu/add16.js +13 -12
- package/dist/ops/webgpu/appendCache.js +79 -54
- package/dist/ops/webgpu/attentionMask.js +41 -25
- package/dist/ops/webgpu/attentionMask32_program.js +34 -26
- package/dist/ops/webgpu/clipScale.js +44 -57
- package/dist/ops/webgpu/concat16.js +96 -111
- package/dist/ops/webgpu/dropout16.js +40 -32
- package/dist/ops/webgpu/gatherSub.js +43 -30
- package/dist/ops/webgpu/gelu.js +88 -82
- package/dist/ops/webgpu/index.js +16 -16
- package/dist/ops/webgpu/matMul16.js +69 -64
- package/dist/ops/webgpu/matMul16_program.js +152 -192
- package/dist/ops/webgpu/mul16.js +13 -12
- package/dist/ops/webgpu/norm2.js +45 -75
- package/dist/ops/webgpu/normRMS.js +25 -33
- package/dist/ops/webgpu/normRMS16_program.js +21 -18
- package/dist/ops/webgpu/normRMS32_program.js +21 -18
- package/dist/ops/webgpu/normRMSGrad.js +125 -184
- package/dist/ops/webgpu/pack16.js +20 -17
- package/dist/ops/webgpu/pack16_program.js +48 -47
- package/dist/ops/webgpu/qkv.js +63 -23
- package/dist/ops/webgpu/rope.js +85 -57
- package/dist/ops/webgpu/scatterSub.js +43 -30
- package/dist/ops/webgpu/slice16.js +66 -61
- package/dist/ops/webgpu/softmax16.js +17 -20
- package/dist/ops/webgpu/softmax16_program.js +34 -18
- package/dist/ops/webgpu/softmax16_subgroup_program.js +40 -45
- package/dist/ops/webgpu/softmax16grad.js +30 -36
- package/dist/ops/webgpu/sub16.js +13 -12
- package/dist/ops/webgpu/sum16.js +28 -37
- package/dist/ops/webgpu/transpose16.js +36 -33
- package/dist/ops/webgpu/transpose16_program.js +40 -39
- package/dist/ops/webgpu/transpose16_shared_program.js +53 -44
- package/dist/ops/webgpu/unpack16.js +49 -37
- package/dist/ops/webgpu/utils/binary_op.js +70 -68
- package/dist/ops/webgpu/utils/deviceInfo.d.ts +1 -1
- package/dist/ops/webgpu/utils/deviceInfo.js +10 -10
- package/dist/ops/webgpu/utils/reductions.js +136 -148
- package/dist/pack16-Ck-spx_F.js +39 -0
- package/dist/patches/webgpu_backend.d.ts +2 -2
- package/dist/patches/webgpu_backend.js +42 -55
- package/dist/patches/webgpu_base.js +21 -33
- package/dist/patches/webgpu_program.js +213 -320
- package/dist/pdf-UoDqCYzz.js +16726 -0
- package/dist/picomatch-3tUnMMbd.js +1063 -0
- package/dist/rope-CbeGlsV8.js +25 -0
- package/dist/selu_util-zkAx5doH.js +24 -0
- package/dist/shared-D1coEFea.js +1314 -0
- package/dist/shared-DOgWaqvL.js +5 -0
- package/dist/slice_util-Dgb3ANWI.js +208 -0
- package/dist/tfjs_backend-BjuQ5FqB.js +614 -0
- package/dist/tokeniser/BaseTokeniser.js +2 -124
- package/dist/tokeniser/CharTokeniser.js +91 -106
- package/dist/tokeniser/bpe.js +163 -166
- package/dist/tokeniser/messages.js +0 -1
- package/dist/tokeniser/type.js +0 -1
- package/dist/training/AdamW.js +127 -137
- package/dist/training/BasicTrainer.d.ts +1 -1
- package/dist/training/BasicTrainer.js +264 -264
- package/dist/training/DatasetBuilder.js +2 -86
- package/dist/training/Evaluator.d.ts +1 -1
- package/dist/training/Evaluator.js +47 -38
- package/dist/training/LRScheduler.js +37 -33
- package/dist/training/PreTrainer.d.ts +2 -2
- package/dist/training/PreTrainer.js +21 -19
- package/dist/training/SFTTrainer.d.ts +2 -2
- package/dist/training/SFTTrainer.js +23 -21
- package/dist/training/loss.js +17 -22
- package/dist/training/orthoGrad.js +9 -9
- package/dist/training/sparseCrossEntropy.js +45 -67
- package/dist/training/tasks/ConversationTask.d.ts +1 -1
- package/dist/training/tasks/ConversationTask.js +36 -38
- package/dist/training/tasks/PretrainingTask.d.ts +1 -1
- package/dist/training/tasks/PretrainingTask.js +41 -46
- package/dist/training/tasks/StartSentenceTask.d.ts +1 -1
- package/dist/training/tasks/StartSentenceTask.js +44 -48
- package/dist/training/tasks/Task.d.ts +1 -1
- package/dist/training/tasks/Task.js +53 -66
- package/dist/training/tasks/splitter.js +17 -20
- package/dist/training/types.d.ts +2 -2
- package/dist/training/types.js +0 -1
- package/dist/training/validation.d.ts +1 -1
- package/dist/training/validation.js +2 -84
- package/dist/utilities/arrayClose.js +15 -19
- package/dist/utilities/datasetID.js +17 -20
- package/dist/utilities/dummy.d.ts +1 -1
- package/dist/utilities/dummy.js +33 -40
- package/dist/utilities/multinomialCPU.js +8 -12
- package/dist/utilities/naming.js +0 -1
- package/dist/utilities/packed.js +10 -12
- package/dist/utilities/parameters.d.ts +1 -1
- package/dist/utilities/parameters.js +32 -51
- package/dist/utilities/performance.js +15 -15
- package/dist/utilities/profile.js +32 -37
- package/dist/utilities/safetensors.js +49 -79
- package/dist/utilities/sentences.d.ts +1 -1
- package/dist/utilities/sentences.js +29 -38
- package/dist/utilities/tokenParse.js +16 -20
- package/dist/utilities/topP.js +11 -12
- package/dist/utilities/waitForModel.d.ts +1 -1
- package/dist/utilities/waitForModel.js +11 -11
- package/dist/utilities/weights.js +37 -42
- package/dist/utilities/yielder.js +6 -6
- package/dist/webgpu-Dt7BMzWz.js +525 -0
- package/dist/webgpu_program-WOyIVMlZ.js +392 -0
- package/dist/webgpu_util-B_F3SShA.js +106 -0
- package/package.json +9 -10
- package/dist/RealDiv-CGwv0liw.js +0 -365
- package/dist/Reshape-BW__R4mZ.js +0 -79
- package/dist/Reshape-CPBkTIH2.js +0 -14
- package/dist/_commonjsHelpers-ByX85dGu.js +0 -33
- package/dist/axis_util-GTVlo58H.js +0 -55
- package/dist/backend_util-GaFarB78.js +0 -425
- package/dist/backend_webgpu-BqASlsbV.js +0 -545
- package/dist/binary_op_util-pKXltfxI.js +0 -192
- package/dist/broadcast_to-eS93CCN_.js +0 -28
- package/dist/clip_by_value-DDA7rrcT.js +0 -12
- package/dist/complex-DI35Q-gW.js +0 -11
- package/dist/complex_util-Yc1A_gV1.js +0 -55
- package/dist/concat-CAQpCret.js +0 -17
- package/dist/concat_util-D18dJ4fD.js +0 -22
- package/dist/data/parquet.d.ts +0 -2
- package/dist/data/parquet.js +0 -17
- package/dist/dataset-CGGp1z9P.js +0 -1124
- package/dist/dropout_util--NxWuYg2.js +0 -27
- package/dist/expand_dims-Bkd1YD5x.js +0 -11
- package/dist/exports_initializers-CYzKLjN7.js +0 -7
- package/dist/floor-BQtb-Azg.js +0 -9
- package/dist/gather-qIqEqaGn.js +0 -9
- package/dist/gelu-B220X1Go.js +0 -26
- package/dist/gpgpu_math-BwvV12df.js +0 -2022
- package/dist/index-CUXkjxiT.js +0 -3516
- package/dist/index-CieiGp4Y.js +0 -349
- package/dist/index-CjOWnMXP.js +0 -7308
- package/dist/index-Cp39cXWe.js +0 -1016
- package/dist/index-D5v913EJ.js +0 -4
- package/dist/index-DmeWGGmS.js +0 -1074
- package/dist/index-DvYrXKkX.js +0 -113
- package/dist/index-Ksja3su6.js +0 -151
- package/dist/index-xuotMAFm.js +0 -118
- package/dist/jszip.min-BZhlzntC.js +0 -2313
- package/dist/kernel_funcs_utils-pq0CK9co.js +0 -306
- package/dist/matMul16-BcVC_E62.js +0 -80
- package/dist/matMulGelu-JNLZqKQp.js +0 -163
- package/dist/mat_mul-DhG0Newp.js +0 -11
- package/dist/mod-CSdCpRjf.js +0 -11
- package/dist/non_max_suppression_impl-B2W7YjZB.js +0 -102
- package/dist/not_equal-hurPF26l.js +0 -64
- package/dist/ones-BytntneX.js +0 -14
- package/dist/ops-CsXeTq1P.js +0 -476
- package/dist/pack16-bqltoUlR.js +0 -39
- package/dist/papaparse.min-C0cScC2i.js +0 -418
- package/dist/parquet-Bqjmp2vo.js +0 -44231
- package/dist/pdf-NIhmP3sq.js +0 -19477
- package/dist/rand_util-CZ7yLoUm.js +0 -50
- package/dist/random_normal-IBRrha8a.js +0 -14
- package/dist/random_width-DN5ZtQkM.js +0 -9796
- package/dist/range-C-CjF-LI.js +0 -10
- package/dist/relu-J_X6MUzx.js +0 -9
- package/dist/reshape-BDOuCSNW.js +0 -9
- package/dist/resize_nearest_neighbor-BojqlfRe.js +0 -150
- package/dist/rope-DcrZM_e6.js +0 -24
- package/dist/scatter_nd_util-ByNJaL6I.js +0 -46
- package/dist/segment_util-Dasb2Zaf.js +0 -43
- package/dist/selu_util-BLhIqRkw.js +0 -44
- package/dist/shared-3agzAqQ_.js +0 -53
- package/dist/shared-CagdqkLh.js +0 -2143
- package/dist/slice-BzS11Qh0.js +0 -12
- package/dist/slice_util-CC35pLmT.js +0 -153
- package/dist/softmax-D4q1LJN7.js +0 -12
- package/dist/split-C2Sj255c.js +0 -9
- package/dist/squeeze-ho4wLUek.js +0 -10
- package/dist/stack-DudVrtmG.js +0 -11
- package/dist/step-BTxPtq1r.js +0 -261
- package/dist/sum-BpiwSWvg.js +0 -11
- package/dist/tensor-BWFldCso.js +0 -8
- package/dist/tensor1d-LMGMIUlr.js +0 -11
- package/dist/tensor2d-BnXMKScO.js +0 -14
- package/dist/tensor4d-C6UCG_u8.js +0 -14
- package/dist/tfjs_backend-BGnG-ppu.js +0 -654
- package/dist/tile-CFy-xTO6.js +0 -11
- package/dist/transpose-9kRxIXWR.js +0 -36
- package/dist/unsorted_segment_sum-DJvk5xnh.js +0 -277
- package/dist/variable-Ck482e3n.js +0 -7
- package/dist/webgpu_program-B4HmApL1.js +0 -525
- package/dist/webgpu_util-DYlGSwOJ.js +0 -64
- package/dist/zeros-DvZpK8s6.js +0 -13
- package/dist/zeros_like-CWjDdwr-.js +0 -721
|
@@ -1,23 +1,27 @@
|
|
|
1
|
-
import {
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
1
|
+
import { Ii as e } from "../../dist-BewPQWjc.js";
|
|
2
|
+
//#region lib/ops/webgl/adamMoments.ts
|
|
3
|
+
var t = class {
|
|
4
|
+
variableNames = ["moments", "gradient"];
|
|
5
|
+
outputShape;
|
|
6
|
+
userCode;
|
|
7
|
+
customUniforms = [{
|
|
8
|
+
name: "beta1",
|
|
9
|
+
type: "float"
|
|
10
|
+
}, {
|
|
11
|
+
name: "beta2",
|
|
12
|
+
type: "float"
|
|
13
|
+
}];
|
|
14
|
+
constructor(e) {
|
|
15
|
+
this.outputShape = e;
|
|
16
|
+
let t = e.length, n = t === 1 ? "int" : `ivec${Math.min(t, 4)}`, r = t === 1 ? "coords" : `coords[${t - 1}]`, i = t === 1 ? "coords" : e.slice(0, -1).map((e, t) => `coords[${t}]`).join(", ");
|
|
17
|
+
this.userCode = `
|
|
14
18
|
void main() {
|
|
15
19
|
float m = getMomentsAtOutCoords();
|
|
16
20
|
${n} coords = getOutputCoords();
|
|
17
21
|
|
|
18
22
|
// Add gradient clipping here
|
|
19
|
-
float g = clamp(getGradient(${
|
|
20
|
-
int which = ${
|
|
23
|
+
float g = clamp(getGradient(${i}), -1.0, 1.0);
|
|
24
|
+
int which = ${r};
|
|
21
25
|
|
|
22
26
|
float beta = which == 0 ? beta1 : beta2;
|
|
23
27
|
float gg = which == 0 ? g : g * g;
|
|
@@ -26,15 +30,15 @@ class i {
|
|
|
26
30
|
setOutput(newM);
|
|
27
31
|
}
|
|
28
32
|
`;
|
|
29
|
-
|
|
30
|
-
}
|
|
31
|
-
function d(o) {
|
|
32
|
-
const { moments: e, gradient: t } = o.inputs, { beta1: n, beta2: a } = o.attrs, r = o.backend, s = new i(e.shape);
|
|
33
|
-
return r.runWebGLProgram(s, [e, t], "float32", [[n], [a]]);
|
|
34
|
-
}
|
|
35
|
-
const g = {
|
|
36
|
-
kernelName: "AdamMoments",
|
|
37
|
-
backendName: "webgl",
|
|
38
|
-
kernelFunc: d
|
|
33
|
+
}
|
|
39
34
|
};
|
|
40
|
-
|
|
35
|
+
function n(e) {
|
|
36
|
+
let { moments: n, gradient: r } = e.inputs, { beta1: i, beta2: a } = e.attrs, o = e.backend, s = new t(n.shape);
|
|
37
|
+
return o.runWebGLProgram(s, [n, r], "float32", [[i], [a]]);
|
|
38
|
+
}
|
|
39
|
+
e({
|
|
40
|
+
kernelName: "AdamMoments",
|
|
41
|
+
backendName: "webgl",
|
|
42
|
+
kernelFunc: n
|
|
43
|
+
});
|
|
44
|
+
//#endregion
|
|
@@ -1,12 +1,21 @@
|
|
|
1
|
-
import {
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
1
|
+
import { Ii as e } from "../../dist-BewPQWjc.js";
|
|
2
|
+
//#region lib/ops/webgl/appendCache.ts
|
|
3
|
+
var t = class {
|
|
4
|
+
variableNames = ["cache", "item"];
|
|
5
|
+
outputShape;
|
|
6
|
+
userCode;
|
|
7
|
+
customUniforms = [{
|
|
8
|
+
name: "cacheT",
|
|
9
|
+
type: "int"
|
|
10
|
+
}];
|
|
11
|
+
constructor(e, t, n, r, i) {
|
|
12
|
+
let a = Math.min(n + 1, i);
|
|
13
|
+
this.outputShape = [
|
|
14
|
+
e,
|
|
15
|
+
t,
|
|
16
|
+
a,
|
|
17
|
+
r
|
|
18
|
+
], this.userCode = `
|
|
10
19
|
void main() {
|
|
11
20
|
ivec4 coords = getOutputCoords(); // [b, h, t, d]
|
|
12
21
|
int b = coords.x;
|
|
@@ -15,7 +24,7 @@ class m {
|
|
|
15
24
|
int d = coords.w;
|
|
16
25
|
|
|
17
26
|
int itemT = 1;
|
|
18
|
-
int maxSize = ${
|
|
27
|
+
int maxSize = ${i};
|
|
19
28
|
int totalT = cacheT + itemT;
|
|
20
29
|
int start = totalT >= maxSize ? 1 : 0;
|
|
21
30
|
|
|
@@ -30,15 +39,15 @@ class m {
|
|
|
30
39
|
setOutput(val);
|
|
31
40
|
}
|
|
32
41
|
`;
|
|
33
|
-
|
|
34
|
-
}
|
|
35
|
-
function d(e) {
|
|
36
|
-
const { cache: t, item: a } = e.inputs, { maxSize: n, pastLen: o } = e.attrs, c = e.backend, s = t.shape[0], r = t.shape[2], i = t.shape[1], h = new m(s, i, r, a.shape[3], n);
|
|
37
|
-
return c.runWebGLProgram(h, [t, a], "float32", [[o]]);
|
|
38
|
-
}
|
|
39
|
-
const l = {
|
|
40
|
-
kernelName: "AppendCache",
|
|
41
|
-
backendName: "webgl",
|
|
42
|
-
kernelFunc: d
|
|
42
|
+
}
|
|
43
43
|
};
|
|
44
|
-
|
|
44
|
+
function n(e) {
|
|
45
|
+
let { cache: n, item: r } = e.inputs, { maxSize: i, pastLen: a } = e.attrs, o = e.backend, s = n.shape[0], c = n.shape[2], l = n.shape[1], u = new t(s, l, c, r.shape[3], i);
|
|
46
|
+
return o.runWebGLProgram(u, [n, r], "float32", [[a]]);
|
|
47
|
+
}
|
|
48
|
+
e({
|
|
49
|
+
kernelName: "AppendCache",
|
|
50
|
+
backendName: "webgl",
|
|
51
|
+
kernelFunc: n
|
|
52
|
+
});
|
|
53
|
+
//#endregion
|
|
@@ -1,15 +1,30 @@
|
|
|
1
|
-
import {
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
1
|
+
import { Ii as e } from "../../dist-BewPQWjc.js";
|
|
2
|
+
//#region lib/ops/webgl/attentionMask.ts
|
|
3
|
+
var t = class {
|
|
4
|
+
variableNames = ["q", "k"];
|
|
5
|
+
outputShape;
|
|
6
|
+
userCode;
|
|
7
|
+
customUniforms = [
|
|
8
|
+
{
|
|
9
|
+
name: "divisor",
|
|
10
|
+
type: "float"
|
|
11
|
+
},
|
|
12
|
+
{
|
|
13
|
+
name: "pastLen",
|
|
14
|
+
type: "int"
|
|
15
|
+
},
|
|
16
|
+
{
|
|
17
|
+
name: "inf",
|
|
18
|
+
type: "float"
|
|
19
|
+
}
|
|
20
|
+
];
|
|
21
|
+
constructor(e, t, n, r, i) {
|
|
22
|
+
this.outputShape = [
|
|
23
|
+
e,
|
|
24
|
+
t,
|
|
25
|
+
n,
|
|
26
|
+
r
|
|
27
|
+
], this.userCode = `
|
|
13
28
|
void main() {
|
|
14
29
|
ivec4 coords = getOutputCoords(); // [batch, nh, t1, t2]
|
|
15
30
|
int b = coords.x;
|
|
@@ -18,7 +33,7 @@ class h {
|
|
|
18
33
|
int t2 = coords.w;
|
|
19
34
|
|
|
20
35
|
float sum = 0.0;
|
|
21
|
-
for (int i = 0; i < ${
|
|
36
|
+
for (int i = 0; i < ${i}; ++i) {
|
|
22
37
|
float qv = getQ(b, h, t1, i);
|
|
23
38
|
float kv = getK(b, h, t2, i);
|
|
24
39
|
sum += qv * kv;
|
|
@@ -31,15 +46,19 @@ class h {
|
|
|
31
46
|
setOutput((t2 > t1 + pastLen) ? inf : scaled);
|
|
32
47
|
}
|
|
33
48
|
`;
|
|
34
|
-
|
|
35
|
-
}
|
|
36
|
-
function l(o) {
|
|
37
|
-
const { q: t, k: e } = o.inputs, { divisor: s, pastLen: n } = o.attrs, a = o.backend, i = t.shape[0], r = t.shape[2], c = e.shape[2], u = t.shape[1], p = t.shape[3], d = new h(i, u, r, c, p);
|
|
38
|
-
return a.runWebGLProgram(d, [t, e], "float32", [[s], [n], [Number.NEGATIVE_INFINITY]]);
|
|
39
|
-
}
|
|
40
|
-
const f = {
|
|
41
|
-
kernelName: "AttentionMask",
|
|
42
|
-
backendName: "webgl",
|
|
43
|
-
kernelFunc: l
|
|
49
|
+
}
|
|
44
50
|
};
|
|
45
|
-
|
|
51
|
+
function n(e) {
|
|
52
|
+
let { q: n, k: r } = e.inputs, { divisor: i, pastLen: a } = e.attrs, o = e.backend, s = n.shape[0], c = n.shape[2], l = r.shape[2], u = n.shape[1], d = n.shape[3], f = new t(s, u, c, l, d);
|
|
53
|
+
return o.runWebGLProgram(f, [n, r], "float32", [
|
|
54
|
+
[i],
|
|
55
|
+
[a],
|
|
56
|
+
[-Infinity]
|
|
57
|
+
]);
|
|
58
|
+
}
|
|
59
|
+
e({
|
|
60
|
+
kernelName: "AttentionMask",
|
|
61
|
+
backendName: "webgl",
|
|
62
|
+
kernelFunc: n
|
|
63
|
+
});
|
|
64
|
+
//#endregion
|
|
@@ -1,11 +1,12 @@
|
|
|
1
|
-
import {
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
1
|
+
import { Ii as e } from "../../dist-BewPQWjc.js";
|
|
2
|
+
//#region lib/ops/webgl/dropout16.ts
|
|
3
|
+
function t(e) {
|
|
4
|
+
let { x: t } = e.inputs;
|
|
5
|
+
return t;
|
|
5
6
|
}
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
};
|
|
11
|
-
|
|
7
|
+
e({
|
|
8
|
+
kernelName: "Dropout16",
|
|
9
|
+
backendName: "webgl",
|
|
10
|
+
kernelFunc: t
|
|
11
|
+
});
|
|
12
|
+
//#endregion
|
|
@@ -1,80 +1,70 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
3
|
-
import {
|
|
4
|
-
|
|
5
|
-
class
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
outputShape;
|
|
27
|
-
userCode;
|
|
28
|
-
customUniforms = [
|
|
29
|
-
{ name: "dropoutRate", type: "float" },
|
|
30
|
-
{ name: "seed", type: "float" }
|
|
31
|
-
];
|
|
32
|
-
constructor(t) {
|
|
33
|
-
this.outputShape = t, this.userCode = `
|
|
34
|
-
float random(ivec4 coords) {
|
|
35
|
-
float x = float(coords.x * 4096 + coords.y * 256 + coords.z * 16 + coords.w);
|
|
36
|
-
return fract(sin(seed + x) * 43758.5453123);
|
|
37
|
-
}
|
|
38
|
-
void main() {
|
|
39
|
-
ivec4 coords = getOutputCoords();
|
|
40
|
-
float numerator = getExp(coords.x, coords.y, coords.z, coords.w);
|
|
41
|
-
float denominator = getSum(coords.x, coords.y, coords.z, coords.w);
|
|
42
|
-
float val = numerator / denominator;
|
|
43
|
-
float keepProb = 1.0 - dropoutRate;
|
|
44
|
-
float rand = random(coords);
|
|
45
|
-
float mask = step(rand, keepProb);
|
|
46
|
-
setOutput(val * mask / keepProb);
|
|
47
|
-
}
|
|
48
|
-
`;
|
|
49
|
-
}
|
|
50
|
-
}
|
|
51
|
-
function L(r) {
|
|
52
|
-
const { inputs: t, attrs: m } = r, { logits: e } = t, { dim: c, dropoutRate: n, seed: u } = m, o = r.backend;
|
|
53
|
-
if (!e)
|
|
54
|
-
throw new Error("Error in softmax: input logits is null");
|
|
55
|
-
const i = P([c], e.shape), d = b({
|
|
56
|
-
inputs: { x: e },
|
|
57
|
-
backend: o,
|
|
58
|
-
attrs: { reductionIndices: i, keepDims: !1 }
|
|
59
|
-
}), f = S(d.shape, i), l = new T(e.shape), s = o.runWebGLProgram(l, [e, d], "float32");
|
|
60
|
-
o.disposeIntermediateTensorInfo(d);
|
|
61
|
-
const p = I({ inputs: { x: s }, backend: o, attrs: { axis: i, keepDims: !1 } }), a = v({ inputs: { x: p }, backend: o, attrs: { shape: f } });
|
|
62
|
-
if (n !== void 0 && n > 0) {
|
|
63
|
-
const h = new C(e.shape), g = o.runWebGLProgram(h, [s, a], "float32", [
|
|
64
|
-
[n],
|
|
65
|
-
[u ?? Math.random() * 1e4]
|
|
66
|
-
]);
|
|
67
|
-
return o.disposeIntermediateTensorInfo(s), o.disposeIntermediateTensorInfo(p), o.disposeIntermediateTensorInfo(a), g;
|
|
68
|
-
}
|
|
69
|
-
const x = k({ inputs: { a: s, b: a }, backend: o });
|
|
70
|
-
return o.disposeIntermediateTensorInfo(s), o.disposeIntermediateTensorInfo(p), o.disposeIntermediateTensorInfo(a), x;
|
|
71
|
-
}
|
|
72
|
-
const E = {
|
|
73
|
-
kernelName: "FusedSoftmax",
|
|
74
|
-
backendName: "webgl",
|
|
75
|
-
kernelFunc: L
|
|
76
|
-
};
|
|
77
|
-
w(E);
|
|
78
|
-
export {
|
|
79
|
-
L as softmax
|
|
1
|
+
import { Ii as e, Tn as t, nc as n } from "../../dist-BewPQWjc.js";
|
|
2
|
+
import { t as r } from "../../Reshape-CABOPB9d.js";
|
|
3
|
+
import { a as i, r as a, t as o } from "../../RealDiv-DBu0FQqT.js";
|
|
4
|
+
//#region lib/ops/webgl/fusedSoftmax.ts
|
|
5
|
+
var s = class {
|
|
6
|
+
variableNames = ["logits", "maxLogits"];
|
|
7
|
+
outputShape;
|
|
8
|
+
userCode;
|
|
9
|
+
constructor(e) {
|
|
10
|
+
this.outputShape = e, this.userCode = "\n void main() {\n ivec4 coords = getOutputCoords(); // [batch, nh, t1, t2]\n int b = coords.x;\n int h = coords.y;\n int t1 = coords.z;\n int t2 = coords.w;\n float x = getLogitsAtOutCoords();\n float maxLogit = getMaxLogits(b, h, t1);\n setOutput(exp(x - maxLogit));\n }\n ";
|
|
11
|
+
}
|
|
12
|
+
}, c = class {
|
|
13
|
+
variableNames = ["exp", "sum"];
|
|
14
|
+
outputShape;
|
|
15
|
+
userCode;
|
|
16
|
+
customUniforms = [{
|
|
17
|
+
name: "dropoutRate",
|
|
18
|
+
type: "float"
|
|
19
|
+
}, {
|
|
20
|
+
name: "seed",
|
|
21
|
+
type: "float"
|
|
22
|
+
}];
|
|
23
|
+
constructor(e) {
|
|
24
|
+
this.outputShape = e, this.userCode = "\n float random(ivec4 coords) {\n float x = float(coords.x * 4096 + coords.y * 256 + coords.z * 16 + coords.w);\n return fract(sin(seed + x) * 43758.5453123);\n }\n void main() {\n ivec4 coords = getOutputCoords();\n float numerator = getExp(coords.x, coords.y, coords.z, coords.w);\n float denominator = getSum(coords.x, coords.y, coords.z, coords.w);\n float val = numerator / denominator;\n float keepProb = 1.0 - dropoutRate;\n float rand = random(coords);\n float mask = step(rand, keepProb);\n setOutput(val * mask / keepProb);\n }\n ";
|
|
25
|
+
}
|
|
80
26
|
};
|
|
27
|
+
function l(e) {
|
|
28
|
+
let { inputs: l, attrs: u } = e, { logits: d } = l, { dim: f, dropoutRate: p, seed: m } = u, h = e.backend;
|
|
29
|
+
if (!d) throw Error("Error in softmax: input logits is null");
|
|
30
|
+
let g = n([f], d.shape), _ = a({
|
|
31
|
+
inputs: { x: d },
|
|
32
|
+
backend: h,
|
|
33
|
+
attrs: {
|
|
34
|
+
reductionIndices: g,
|
|
35
|
+
keepDims: !1
|
|
36
|
+
}
|
|
37
|
+
}), v = t(_.shape, g), y = new s(d.shape), b = h.runWebGLProgram(y, [d, _], "float32");
|
|
38
|
+
h.disposeIntermediateTensorInfo(_);
|
|
39
|
+
let x = i({
|
|
40
|
+
inputs: { x: b },
|
|
41
|
+
backend: h,
|
|
42
|
+
attrs: {
|
|
43
|
+
axis: g,
|
|
44
|
+
keepDims: !1
|
|
45
|
+
}
|
|
46
|
+
}), S = r({
|
|
47
|
+
inputs: { x },
|
|
48
|
+
backend: h,
|
|
49
|
+
attrs: { shape: v }
|
|
50
|
+
});
|
|
51
|
+
if (p !== void 0 && p > 0) {
|
|
52
|
+
let e = new c(d.shape), t = h.runWebGLProgram(e, [b, S], "float32", [[p], [m ?? Math.random() * 1e4]]);
|
|
53
|
+
return h.disposeIntermediateTensorInfo(b), h.disposeIntermediateTensorInfo(x), h.disposeIntermediateTensorInfo(S), t;
|
|
54
|
+
}
|
|
55
|
+
let C = o({
|
|
56
|
+
inputs: {
|
|
57
|
+
a: b,
|
|
58
|
+
b: S
|
|
59
|
+
},
|
|
60
|
+
backend: h
|
|
61
|
+
});
|
|
62
|
+
return h.disposeIntermediateTensorInfo(b), h.disposeIntermediateTensorInfo(x), h.disposeIntermediateTensorInfo(S), C;
|
|
63
|
+
}
|
|
64
|
+
e({
|
|
65
|
+
kernelName: "FusedSoftmax",
|
|
66
|
+
backendName: "webgl",
|
|
67
|
+
kernelFunc: l
|
|
68
|
+
});
|
|
69
|
+
//#endregion
|
|
70
|
+
export { l as softmax };
|
|
@@ -1,27 +1,28 @@
|
|
|
1
|
-
import {
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
float logit = getLogits(coords, index);
|
|
13
|
-
|
|
14
|
-
}
|
|
15
|
-
`;
|
|
16
|
-
}
|
|
17
|
-
}
|
|
18
|
-
function i(t) {
|
|
19
|
-
const { logits: e, labels: o, values: s } = t.inputs, a = t.backend, r = o.shape[0], n = new u(r);
|
|
20
|
-
return a.runWebGLProgram(n, [o, e, s], "float32");
|
|
21
|
-
}
|
|
22
|
-
const c = {
|
|
23
|
-
kernelName: "EfficientGatherSub",
|
|
24
|
-
backendName: "webgl",
|
|
25
|
-
kernelFunc: i
|
|
1
|
+
import { Ii as e } from "../../dist-BewPQWjc.js";
|
|
2
|
+
//#region lib/ops/webgl/gatherSub.ts
|
|
3
|
+
var t = class {
|
|
4
|
+
variableNames = [
|
|
5
|
+
"labels",
|
|
6
|
+
"logits",
|
|
7
|
+
"values"
|
|
8
|
+
];
|
|
9
|
+
outputShape;
|
|
10
|
+
userCode;
|
|
11
|
+
constructor(e) {
|
|
12
|
+
this.outputShape = [e], this.userCode = "\n void main() {\n int coords = getOutputCoords();\n int index = int(getLabelsAtOutCoords());\n float val = getValuesAtOutCoords();\n float logit = getLogits(coords, index);\n setOutput(val - logit);\n }\n ";
|
|
13
|
+
}
|
|
26
14
|
};
|
|
27
|
-
|
|
15
|
+
function n(e) {
|
|
16
|
+
let { logits: n, labels: r, values: i } = e.inputs, a = e.backend, o = r.shape[0], s = new t(o);
|
|
17
|
+
return a.runWebGLProgram(s, [
|
|
18
|
+
r,
|
|
19
|
+
n,
|
|
20
|
+
i
|
|
21
|
+
], "float32");
|
|
22
|
+
}
|
|
23
|
+
e({
|
|
24
|
+
kernelName: "EfficientGatherSub",
|
|
25
|
+
backendName: "webgl",
|
|
26
|
+
kernelFunc: n
|
|
27
|
+
});
|
|
28
|
+
//#endregion
|
package/dist/ops/webgl/gelu.js
CHANGED
|
@@ -1,50 +1,48 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
3
|
-
|
|
1
|
+
import { Ii as e } from "../../dist-BewPQWjc.js";
|
|
2
|
+
import { _ as t, i as n } from "../../kernel_funcs_utils-HiXOOx3f.js";
|
|
3
|
+
//#region lib/ops/webgl/gelu.ts
|
|
4
|
+
var r = .7978845608028654, i = .044715, a = n({ opSnippet: t + `
|
|
4
5
|
float x3 = x * x * x;
|
|
5
|
-
float inner = x + ${
|
|
6
|
-
inner = ${
|
|
6
|
+
float inner = x + ${i} * x3;
|
|
7
|
+
inner = ${r} * inner;
|
|
7
8
|
inner = abs(inner) > 15.0 ? sign(inner) : tanh(inner);
|
|
8
9
|
inner = 0.5 * (1.0 + inner);
|
|
9
10
|
inner = x * inner;
|
|
10
11
|
return inner;
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
class
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
this.outputShape = n, this.userCode = `
|
|
12
|
+
` });
|
|
13
|
+
e({
|
|
14
|
+
kernelName: "Gelu",
|
|
15
|
+
backendName: "webgl",
|
|
16
|
+
kernelFunc: a
|
|
17
|
+
});
|
|
18
|
+
var o = class {
|
|
19
|
+
variableNames = ["dy", "x"];
|
|
20
|
+
outputShape;
|
|
21
|
+
userCode;
|
|
22
|
+
constructor(e) {
|
|
23
|
+
this.outputShape = e, this.userCode = `
|
|
24
24
|
void main() {
|
|
25
25
|
float dy = getDyAtOutCoords();
|
|
26
26
|
float x = getXAtOutCoords();
|
|
27
27
|
float x2 = x * x;
|
|
28
28
|
float x3 = x2 * x;
|
|
29
|
-
float u = ${
|
|
29
|
+
float u = ${r} * (x + ${i} * x3);
|
|
30
30
|
float t = abs(u) > 15.0 ? sign(u) : tanh(u);
|
|
31
31
|
float sech2 = 1.0 - t * t;
|
|
32
|
-
float du_dx = ${
|
|
32
|
+
float du_dx = ${r} * (1.0 + 3.0 * ${i} * x2);
|
|
33
33
|
float dgelu = 0.5 * (1.0 + t) + 0.5 * x * sech2 * du_dx;
|
|
34
34
|
setOutput(dy * dgelu);
|
|
35
35
|
}`;
|
|
36
|
-
|
|
37
|
-
}
|
|
38
|
-
function g(e) {
|
|
39
|
-
const { dy: n, x: o } = e.inputs, u = e.backend, l = new f(o.shape);
|
|
40
|
-
return u.runWebGLProgram(l, [n, o], "float32");
|
|
41
|
-
}
|
|
42
|
-
const p = {
|
|
43
|
-
kernelName: "GeluGrad",
|
|
44
|
-
backendName: "webgl",
|
|
45
|
-
kernelFunc: g
|
|
46
|
-
};
|
|
47
|
-
a(p);
|
|
48
|
-
export {
|
|
49
|
-
d as gelu
|
|
36
|
+
}
|
|
50
37
|
};
|
|
38
|
+
function s(e) {
|
|
39
|
+
let { dy: t, x: n } = e.inputs, r = e.backend, i = new o(n.shape);
|
|
40
|
+
return r.runWebGLProgram(i, [t, n], "float32");
|
|
41
|
+
}
|
|
42
|
+
e({
|
|
43
|
+
kernelName: "GeluGrad",
|
|
44
|
+
backendName: "webgl",
|
|
45
|
+
kernelFunc: s
|
|
46
|
+
});
|
|
47
|
+
//#endregion
|
|
48
|
+
export { a as gelu };
|
package/dist/ops/webgl/log.js
CHANGED
|
@@ -1,23 +1,14 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
3
|
-
import {
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
result.g = isNaN.g ? x.g : (x.g < 0.0 ? NAN : result.g)
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
opSnippet: a,
|
|
16
|
-
packedOpSnippet: t,
|
|
17
|
-
cpuKernelImpl: l
|
|
18
|
-
}), o = {
|
|
19
|
-
kernelName: r,
|
|
20
|
-
backendName: "webgl",
|
|
21
|
-
kernelFunc: n
|
|
22
|
-
};
|
|
23
|
-
e(o);
|
|
1
|
+
import { Ii as e } from "../../dist-BewPQWjc.js";
|
|
2
|
+
import { y as t } from "../../shared-DOgWaqvL.js";
|
|
3
|
+
import { i as n, t as r } from "../../kernel_funcs_utils-HiXOOx3f.js";
|
|
4
|
+
//#region lib/ops/webgl/log.ts
|
|
5
|
+
e({
|
|
6
|
+
kernelName: "Log",
|
|
7
|
+
backendName: "webgl",
|
|
8
|
+
kernelFunc: n({
|
|
9
|
+
opSnippet: r + "\n return x < 0.0 ? NAN : log(x);\n",
|
|
10
|
+
packedOpSnippet: "\n vec4 result = log(x);\n bvec4 isNaN = isnan(x);\n result.r = isNaN.r ? x.r : (x.r < 0.0 ? NAN : result.r);\n result.g = isNaN.g ? x.g : (x.g < 0.0 ? NAN : result.g);\n result.b = isNaN.b ? x.b : (x.b < 0.0 ? NAN : result.b);\n result.a = isNaN.a ? x.a : (x.a < 0.0 ? NAN : result.a);\n return result;\n",
|
|
11
|
+
cpuKernelImpl: t
|
|
12
|
+
})
|
|
13
|
+
});
|
|
14
|
+
//#endregion
|
|
@@ -1,45 +1,37 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
3
|
-
import { matMulMul as
|
|
4
|
-
import { matMulGelu as
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
}
|
|
32
|
-
else if (n) {
|
|
33
|
-
const s = v(e, n);
|
|
34
|
-
return e.dispose(), s;
|
|
35
|
-
} else
|
|
36
|
-
return e;
|
|
37
|
-
}
|
|
38
|
-
throw new Error("Not implemented: matMul16 with packed inputs");
|
|
1
|
+
import { Ii as e, Wr as t, _n as n, br as r, gr as i, hn as a, lt as o, w as s } from "../../dist-BewPQWjc.js";
|
|
2
|
+
import { isPackedTensor as c } from "../../utilities/packed.js";
|
|
3
|
+
import { matMulMul as l } from "../matMulMul.js";
|
|
4
|
+
import { matMulGelu as u } from "../matMulGelu.js";
|
|
5
|
+
//#region lib/ops/webgl/matMul16.ts
|
|
6
|
+
function d(e) {
|
|
7
|
+
let { A: d, B: f } = e.inputs, { transposeA: p, transposeB: m, scale: h, activation: g, scaleA: _, scaleB: v, forceOutputShape: y, perm: b } = e.attrs, x = !c(d), S = !c(f);
|
|
8
|
+
if (x && S) {
|
|
9
|
+
let e = _ === void 0 ? d : t(d, n(_)), c = v === void 0 ? f : t(f, n(v)), x;
|
|
10
|
+
if (h !== void 0) x = l(e, c, n(h), p, m);
|
|
11
|
+
else if (g === "gelu") x = u(e, c);
|
|
12
|
+
else if (g === "relu2") {
|
|
13
|
+
let t = i(e, c, p, m), n = o(t);
|
|
14
|
+
t.dispose(), x = a(n), n.dispose();
|
|
15
|
+
} else x = g === "relu" ? o(i(e, c, p, m)) : i(e, c, p, m);
|
|
16
|
+
if (b) if (y) {
|
|
17
|
+
let e = r(x, y);
|
|
18
|
+
x.dispose();
|
|
19
|
+
let t = s(e, b);
|
|
20
|
+
return e.dispose(), t;
|
|
21
|
+
} else {
|
|
22
|
+
let e = s(x, b);
|
|
23
|
+
return x.dispose(), e;
|
|
24
|
+
}
|
|
25
|
+
else if (y) {
|
|
26
|
+
let e = r(x, y);
|
|
27
|
+
return x.dispose(), e;
|
|
28
|
+
} else return x;
|
|
29
|
+
}
|
|
30
|
+
throw Error("Not implemented: matMul16 with packed inputs");
|
|
39
31
|
}
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
};
|
|
45
|
-
|
|
32
|
+
e({
|
|
33
|
+
kernelName: "MatMul16",
|
|
34
|
+
backendName: "webgl",
|
|
35
|
+
kernelFunc: d
|
|
36
|
+
});
|
|
37
|
+
//#endregion
|