@genai-fi/nanogpt 1.4.0 → 1.5.0

package/dist/matMul16-DGtlziMn.js

@@ -1,81 +0,0 @@
-import { Fi as e, ii as t } from "./dist-Da20xy8E.js";
-import { t as n } from "./gelu-B6637MKa.js";
-import { isPackedTensor as r } from "./utilities/packed.js";
-import { transpose16 as i } from "./ops/transpose16.js";
-import { reshape16 as a } from "./ops/reshape16.js";
-import { mul16 as o } from "./ops/mul16.js";
-import "./ops/webgl/matMul16.js";
-import "./ops/cpu/matMul16.js";
-import { t as s } from "./pack16-BhuXNUS7.js";
-//#region lib/ops/grads/matMul16.ts
-var c = {
-	kernelName: "MatMul16",
-	inputsToSave: ["A", "B"],
-	outputsToSave: [],
-	gradFunc: (e, t, r) => {
-		let [s, c] = t;
-		if (Array.isArray(e)) throw Error("Expected dy to be a single Tensor");
-		let u = e, { transposeA: p, transposeB: m, scale: h, activation: g, originalShape: _, perm: v } = r;
-		if (v && _) {
-			let e = Array(v.length);
-			for (let t = 0; t < v.length; ++t) e[v[t]] = t;
-			let t = u;
-			u = i(u, e), t.dispose();
-		}
-		if (_) {
-			let e = u;
-			u = a(u, _), e.dispose();
-		}
-		if (g === "gelu") {
-			let e = u, t = l(s, c, p, m);
-			u = n(e, t), e.dispose(), t.dispose();
-		} else if (g === "relu2") {
-			let e = u, t = l(s, c, p, m, {
-				activation: "relu",
-				scale: 2
-			});
-			u = o(e, t), e.dispose(), t.dispose();
-		}
-		if (!p && !m) return {
-			A: () => h === void 0 ? l(u, c, !1, !0) : d(u, c, h, !1, !0),
-			B: () => h === void 0 ? l(s, u, !0, !1) : f(s, u, h, !0, !1)
-		};
-		if (!p && m) return {
-			A: () => h === void 0 ? l(u, c, !1, !1) : d(u, c, h, !1, !1),
-			B: () => h === void 0 ? l(s, u, !0, !1) : f(s, u, h, !0, !1)
-		};
-		if (p && !m) return {
-			A: () => h === void 0 ? l(c, u, !1, !0) : f(c, u, h, !1, !0),
-			B: () => h === void 0 ? l(s, u, !1, !1) : f(s, u, h, !1, !1)
-		};
-		throw Error("Gradient for transposeA=true and transposeB=true is not supported yet.");
-	}
-};
-e(c);
-//#endregion
-//#region lib/ops/matMul16.ts
-function l(e, n, i = !1, a = !1, o = {}) {
-	let c = r(e), l = r(n), u = c || l, d = !u || c ? e : s(e), f = !u || l ? n : s(n), p = t().runKernel("MatMul16", {
-		A: d,
-		B: f
-	}, {
-		transposeA: i,
-		transposeB: a,
-		...o
-	});
-	return u && !c && d.dispose(), u && !l && f.dispose(), p;
-}
-function u(e, t, n, r = !1, i = !1) {
-	return l(e, t, r, i, { scale: n });
-}
-function d(e, t, n, r = !1, i = !1) {
-	return l(e, t, r, i, { scaleA: n });
-}
-function f(e, t, n, r = !1, i = !1) {
-	return l(e, t, r, i, { scaleB: n });
-}
-function p(e, t, n = !1, r = !1) {
-	return l(e, t, n, r, { activation: "gelu" });
-}
-//#endregion
-export { u as a, f as i, p as n, c as o, d as r, l as t };