@jax-js/jax 0.1.5 → 0.1.6

package/dist/{backend-DziQSaoQ.cjs → backend-D7s-Retx.cjs}

@@ -151,6 +151,19 @@ function normalizeAxis(axis, ndim) {
 		return sorted(seen);
 	}
 }
+/** Check for an array of integers with no duplicates. */
+function checkInts(indices) {
+	if (typeof indices === "number") {
+		if (!Number.isInteger(indices)) throw new TypeError(`Expected integer index, got ${indices}`);
+	} else {
+		const seen = /* @__PURE__ */ new Set();
+		for (const i of indices) {
+			if (!Number.isInteger(i)) throw new TypeError(`Expected integer indices, got ${i}`);
+			if (seen.has(i)) throw new Error(`Duplicate index ${i} passed to function`);
+			seen.add(i);
+		}
+	}
+}
 function range(start, stop, step = 1) {
 	if (stop === void 0) {
 		stop = start;
@@ -2306,10 +2319,10 @@ function tuneWebgpu(kernel) {
 			upcastedAxis.add(choices[0][2]);
 		} else break;
 	}
-	if (/chrome/i.test(navigator.userAgent) && dim.reduce < dim.unroll && (prod(dim.st.shape.slice(dim.unroll)) <= 4 || dim.unroll === dim.upcast && prod(dim.st.shape.slice(dim.upcast)) < 64)) {
+	if (!/Mobi|Android/i.test(navigator.userAgent) && dim.reduce < dim.unroll && (prod(dim.st.shape.slice(dim.unroll)) <= 4 || dim.unroll === dim.upcast && prod(dim.st.shape.slice(dim.upcast)) < 64)) {
 		const s = dim.st.shape[dim.unroll - 1];
 		if (0 < s && s <= 32) dim.applyUnroll(dim.reduce, s);
-		else for (const splits of [4]) if (s % splits === 0) {
+		else for (const splits of [8, 4]) if (s % splits === 0) {
 			dim.applyUnroll(dim.unroll - 1, splits);
 			break;
 		}
@@ -4240,7 +4253,7 @@ async function createBackend(device) {
 		if (!navigator.gpu) return null;
 		const adapter = await navigator.gpu.requestAdapter({ powerPreference: "high-performance" });
 		if (!adapter) return null;
-		const { WebGPUBackend } = await Promise.resolve().then(() => require("./webgpu-Db2JrNBr.cjs"));
+		const { WebGPUBackend } = await Promise.resolve().then(() => require("./webgpu-rraa6dfz.cjs"));
 		const importantLimits = [
 			"maxBufferSize",
 			"maxComputeInvocationsPerWorkgroup",
@@ -4278,7 +4291,7 @@ async function createBackend(device) {
 		});
 		if (!gl) return null;
 		if (!gl.getExtension("EXT_color_buffer_float")) return null;
-		const { WebGLBackend } = await Promise.resolve().then(() => require("./webgl-ClIYb8jP.cjs"));
+		const { WebGLBackend } = await Promise.resolve().then(() => require("./webgl-CyfzNW8T.cjs"));
 		return new WebGLBackend(gl);
 	} else throw new Error(`Backend not found: ${device}`);
 }
@@ -4446,6 +4459,12 @@ Object.defineProperty(exports, 'checkAxis', {
     return checkAxis;
   }
 });
+Object.defineProperty(exports, 'checkInts', {
+  enumerable: true,
+  get: function () {
+    return checkInts;
+  }
+});
 Object.defineProperty(exports, 'deepEqual', {
   enumerable: true,
   get: function () {

package/dist/{backend-DaqL-MNz.js → backend-Dx6Ob2D1.js}

@@ -150,6 +150,19 @@ function normalizeAxis(axis, ndim) {
 		return sorted(seen);
 	}
 }
+/** Check for an array of integers with no duplicates. */
+function checkInts(indices) {
+	if (typeof indices === "number") {
+		if (!Number.isInteger(indices)) throw new TypeError(`Expected integer index, got ${indices}`);
+	} else {
+		const seen = /* @__PURE__ */ new Set();
+		for (const i of indices) {
+			if (!Number.isInteger(i)) throw new TypeError(`Expected integer indices, got ${i}`);
+			if (seen.has(i)) throw new Error(`Duplicate index ${i} passed to function`);
+			seen.add(i);
+		}
+	}
+}
 function range(start, stop, step = 1) {
 	if (stop === void 0) {
 		stop = start;
@@ -2305,10 +2318,10 @@ function tuneWebgpu(kernel) {
 			upcastedAxis.add(choices[0][2]);
 		} else break;
 	}
-	if (/chrome/i.test(navigator.userAgent) && dim.reduce < dim.unroll && (prod(dim.st.shape.slice(dim.unroll)) <= 4 || dim.unroll === dim.upcast && prod(dim.st.shape.slice(dim.upcast)) < 64)) {
+	if (!/Mobi|Android/i.test(navigator.userAgent) && dim.reduce < dim.unroll && (prod(dim.st.shape.slice(dim.unroll)) <= 4 || dim.unroll === dim.upcast && prod(dim.st.shape.slice(dim.upcast)) < 64)) {
 		const s = dim.st.shape[dim.unroll - 1];
 		if (0 < s && s <= 32) dim.applyUnroll(dim.reduce, s);
-		else for (const splits of [4]) if (s % splits === 0) {
+		else for (const splits of [8, 4]) if (s % splits === 0) {
 			dim.applyUnroll(dim.unroll - 1, splits);
 			break;
 		}
@@ -4239,7 +4252,7 @@ async function createBackend(device) {
 		if (!navigator.gpu) return null;
 		const adapter = await navigator.gpu.requestAdapter({ powerPreference: "high-performance" });
 		if (!adapter) return null;
-		const { WebGPUBackend } = await import("./webgpu-Dh7k9io0.js");
+		const { WebGPUBackend } = await import("./webgpu-C-VfevQW.js");
 		const importantLimits = [
 			"maxBufferSize",
 			"maxComputeInvocationsPerWorkgroup",
@@ -4277,7 +4290,7 @@ async function createBackend(device) {
 		});
 		if (!gl) return null;
 		if (!gl.getExtension("EXT_color_buffer_float")) return null;
-		const { WebGLBackend } = await import("./webgl-RSuZKvgc.js");
+		const { WebGLBackend } = await import("./webgl-CLLvzJlO.js");
 		return new WebGLBackend(gl);
 	} else throw new Error(`Backend not found: ${device}`);
 }
@@ -4313,4 +4326,4 @@ var UnsupportedRoutineError = class extends Error {
 };
 
 //#endregion
-export { AluExp, AluGroup, AluOp, AluVar, DEBUG, DType, Executable, FpHash, Kernel, PPrint, Reduction, Routine, Routines, ShapeTracker, SlotError, UnsupportedOpError, UnsupportedRoutineError, accessorAluExp, accessorGlobal, assertNonNull, byteWidth, checkAxis, deepEqual, defaultDevice, devices, dtypedArray, dtypedJsArray, findPow2, generalBroadcast, getBackend, init, invertPermutation, isFloatDtype, isNumberPair, isPermutation, mapSetUnion, normalizeAxis, partitionList, prod, promoteTypes, range, recursiveFlatten, rep, runWithCache, setDebug, strip1, toposort, tuneNullopt, tuneWebgpu, unravelAlu, unzip2, zip, zipn };
+export { AluExp, AluGroup, AluOp, AluVar, DEBUG, DType, Executable, FpHash, Kernel, PPrint, Reduction, Routine, Routines, ShapeTracker, SlotError, UnsupportedOpError, UnsupportedRoutineError, accessorAluExp, accessorGlobal, assertNonNull, byteWidth, checkAxis, checkInts, deepEqual, defaultDevice, devices, dtypedArray, dtypedJsArray, findPow2, generalBroadcast, getBackend, init, invertPermutation, isFloatDtype, isNumberPair, isPermutation, mapSetUnion, normalizeAxis, partitionList, prod, promoteTypes, range, recursiveFlatten, rep, runWithCache, setDebug, strip1, toposort, tuneNullopt, tuneWebgpu, unravelAlu, unzip2, zip, zipn };