@genai-fi/nanogpt 0.10.2 → 0.10.3

package/dist/patches/webgpu_program.js

@@ -1,10 +1,9 @@
-import { h as z } from "../index-D6Q1lPZO.js";
-import { g as F, d as b, a as x, b as g, s as O, t as l, c as y } from "../webgpu_program-DzaQiqel.js";
-import { e as _, a as A, b as L } from "../tensor-CzmOBsdf.js";
+import { j as z, l as F, n as O, o as _ } from "../index-DOvlwCh-.js";
+import { g as A, d as b, a as x, b as g, s as L, t as l, c as y } from "../webgpu_program-DuOXPQol.js";
 var N = /* @__PURE__ */ ((s) => (s[s.FROM_PIXELS = 0] = "FROM_PIXELS", s[s.DRAW = 1] = "DRAW", s))(N || {});
-const K = (s, t, e, o, u) => {
-  const a = { dtype: o.dtype, shape: o.shape }, n = D(e, a, t), r = s.createShaderModule({ code: n, label: t.constructor.name });
-  let d = _().get("WEBGPU_PRINT_SHADER");
+const H = (s, t, e, o, u) => {
+  const a = { dtype: o.dtype, shape: o.shape }, n = j(e, a, t), r = s.createShaderModule({ code: n, label: t.constructor.name });
+  let d = z().get("WEBGPU_PRINT_SHADER");
   if (d !== "") {
     d = d.toLowerCase();
     const f = d.split(",");
@@ -20,7 +19,7 @@ const K = (s, t, e, o, u) => {
     layout: "auto"
   });
 };
-function D(s, t, e) {
+function j(s, t, e) {
   const o = [], u = e.workgroupSize[0] * e.workgroupSize[1] * e.workgroupSize[2];
   if (e.outputComponent = e.outputComponent ? e.outputComponent : 1, o.push(`
 
@@ -65,7 +64,7 @@ function D(s, t, e) {
 `),
       x(t.shape),
       e.getUserCode(),
-      m(I, e)
+      k(I, e)
     ].join(`
 `);
   }
@@ -94,7 +93,7 @@ function D(s, t, e) {
     e.subgroups ? "enable subgroups;" : "",
     C,
     o.join(`
-`) + T,
+`) + D,
     x(t.shape),
     f,
     W(t.shape.length)
@@ -113,7 +112,7 @@ function D(s, t, e) {
 `);
   $.push(c), $.push(e.getUserCode());
   const p = w(e);
-  return $.push(m(p, e)), $.join(`
+  return $.push(k(p, e)), $.join(`
 `);
 }
 const C = `
@@ -166,12 +165,12 @@ const C = `
     let floatToUint: vec4<u32> = bitcast<vec4<u32>>(val);
     return (floatToUint & vec4<u32>(0x7fffffffu)) > vec4<u32>(0x7f800000u);
   }
-`, T = `
+`, D = `
   fn isinf(val: f32) -> bool {
     return abs(val) == uniforms.INFINITY;
   }
 `;
-function j(s, t) {
+function T(s, t) {
   const e = s.name, o = s.shape.length, u = g(o), a = "get" + e.charAt(0).toUpperCase() + e.slice(1), n = ["d0", "d1", "d2", "d3", "d4", "d5"].slice(0, o), r = n.map((v) => `${v} : i32`).join(", ");
   if (o < 1)
     return `
@@ -190,7 +189,7 @@ function j(s, t) {
 }
 function E(s, t, e, o) {
   const u = s.name, a = u.charAt(0).toUpperCase() + u.slice(1), n = "get" + a + "ByOutput", r = s.shape.length, d = t.length, f = g(d);
-  if (L(s.shape, t) && o)
+  if (O(s.shape, t) && o)
     return `
     fn ${n}Index(globalIndex : i32) -> ${l(e)} {
       return ${l(e)}(${u}[globalIndex]);
@@ -200,7 +199,7 @@ function E(s, t, e, o) {
       return ${l(e)}(${u}[${d > 1 ? "getOutputIndexFromCoords(coords)" : "coords"}${e === 1 ? "" : ` / ${e}`}]);
     }
     `;
-  const v = z(s.shape, t), $ = d - r;
+  const v = _(s.shape, t), $ = d - r;
   let c = "";
   if (r === 0)
     return `
@@ -218,7 +217,7 @@ function E(s, t, e, o) {
   if (d < 2 && r > 0)
     p = "coords";
   else if (d > 1) {
-    const h = g(r), I = s.shape.map((M, k) => `coords.${y(k + $)}`).join(", ");
+    const h = g(r), I = s.shape.map((M, m) => `coords.${y(m + $)}`).join(", ");
     p = `${h}(${I})`;
   } else
     p = "coords";
@@ -238,7 +237,7 @@ function E(s, t, e, o) {
 `;
 }
 function U(s, t, e, o) {
-  let u = j(s, e);
+  let u = T(s, e);
   return s.shape.length <= t.length && (u += E(s, t, e, o)), u;
 }
 function B(s, t) {
@@ -259,7 +258,7 @@ function B(s, t) {
       if (p.length === 1)
         r += `let d${p[0]} = i32(globalId[${c}]);`;
       else {
-        const S = O(p, "uniforms.outShape");
+        const S = L(p, "uniforms.outShape");
         r += `var index${c} = i32(globalId[${c}]);`;
         for (let i = 0; i < S.length; i++)
           r += `let d${p[i]} = index${c} / ${S[i]};`, i === S.length - 1 ? r += `let d${p[i + 1]} = index${c} - d${p[i]} * ${S[i]};` : r += `index${c} = index${c} - d${p[i]} * ${S[i]};`;
@@ -331,7 +330,7 @@ function W(s) {
         `;
       break;
     default:
-      A(!1, () => `Unsupported ${s}D shape`);
+      F(!1, () => `Unsupported ${s}D shape`);
       break;
   }
   return t;
@@ -373,9 +372,9 @@ function P(s) {
 function w(s) {
   return !(s.dispatchLayout.hasOwnProperty("y") && s.dispatchLayout.y?.length !== 0 || s.dispatchLayout.hasOwnProperty("z") && s.dispatchLayout.z?.length !== 0);
 }
-function m(s, t) {
+function k(s, t) {
   return `
-     ${F(t)}
+     ${A(t)}
       fn _start(@builtin(local_invocation_id) LocalId : vec3<u32>,
                 @builtin(global_invocation_id) GlobalId : vec3<u32>,
                 @builtin(local_invocation_index) LocalIndex: u32,
@@ -396,6 +395,6 @@ function m(s, t) {
 }
 export {
   N as PixelsOpType,
-  K as compileProgram,
-  m as getStartHeaderString
+  H as compileProgram,
+  k as getStartHeaderString
 };