typegpu 0.11.1 → 0.11.3

package/_virtual/{_rolldown/runtime.js → rolldown_runtime.js}

@@ -1,13 +1,13 @@
-//#region \0rolldown/runtime.js
+//#region rolldown:runtime
 var __defProp = Object.defineProperty;
-var __exportAll = (all, no_symbols) => {
+var __export = (all) => {
 	let target = {};
 	for (var name in all) __defProp(target, name, {
 		get: all[name],
 		enumerable: true
 	});
-	if (!no_symbols) __defProp(target, Symbol.toStringTag, { value: "Module" });
 	return target;
 };
+
 //#endregion
-export { __exportAll };
+export { __export };

package/builtin.js

@@ -3,6 +3,7 @@ import { bool, f32, u32 } from "./data/numeric.js";
 import { vec3u, vec4f } from "./data/vector.js";
 import { attribute } from "./data/attributes.js";
 import { arrayOf } from "./data/array.js";
+
 //#region src/builtin.ts
 function defineBuiltin(dataType, value) {
 	return attribute(dataType, {
@@ -31,5 +32,6 @@ const builtin = {
 	subgroupId: defineBuiltin(u32, "subgroup_id"),
 	numSubgroups: defineBuiltin(u32, "num_subgroups")
 };
+
 //#endregion
-export { builtin };
+export { builtin };

package/common/fullScreenTriangle.d.ts

@@ -1,6 +1,9 @@
 import { TgpuVertexFn } from "../core/function/tgpuVertexFn.js";
 import { Vec2f } from "../data/wgslTypes.js";
+import "../indexNamedExports.js";
+
 //#region src/common/fullScreenTriangle.d.ts
+
 /**
  * A vertex function that defines a single full-screen triangle out
  * of three points.
@@ -17,6 +20,8 @@ import { Vec2f } from "../data/wgslTypes.js";
  * pipeline.draw(3);
  * ```
  */
-declare const fullScreenTriangle: TgpuVertexFn;
+declare const fullScreenTriangle: TgpuVertexFn<{}, {
+  uv: Vec2f;
+}>;
 //#endregion
 export { fullScreenTriangle };

package/common/fullScreenTriangle.js

@@ -1,6 +1,7 @@
 import { vec2f } from "../data/vector.js";
 import { builtin } from "../builtin.js";
 import { vertexFn } from "../core/function/tgpuVertexFn.js";
+
 //#region src/common/fullScreenTriangle.ts
 /**
 * A vertex function that defines a single full-screen triangle out
@@ -30,5 +31,6 @@ const fullScreenTriangle = vertexFn({
 
   return Out(vec4f(pos[in.vertexIndex], 0, 1), uv[in.vertexIndex]);
 }`;
+
 //#endregion
-export { fullScreenTriangle };
+export { fullScreenTriangle };

package/common/index.js

@@ -1,10 +1,12 @@
-import { __exportAll } from "../_virtual/_rolldown/runtime.js";
+import { __export } from "../_virtual/rolldown_runtime.js";
 import { fullScreenTriangle } from "./fullScreenTriangle.js";
 import { writeSoA } from "./writeSoA.js";
+
 //#region src/common/index.ts
-var common_exports = /* @__PURE__ */ __exportAll({
+var common_exports = /* @__PURE__ */ __export({
 	fullScreenTriangle: () => fullScreenTriangle,
 	writeSoA: () => writeSoA
 });
+
 //#endregion
-export { common_exports, fullScreenTriangle, writeSoA };
+export { common_exports, fullScreenTriangle, writeSoA };

package/common/writeSoA.js

@@ -5,6 +5,7 @@ import { alignmentOf } from "../data/alignmentOf.js";
 import { roundUp } from "../mathUtils.js";
 import { sizeOf } from "../data/sizeOf.js";
 import { offsetsForProps } from "../data/offsets.js";
+
 //#region src/common/writeSoA.ts
 function packedSchemaOf(schema) {
 	const unpackedSchema = undecorate(schema);
@@ -86,5 +87,6 @@ function writeSoA(buffer, data, options) {
 		endOffset
 	});
 }
+
 //#endregion
-export { writeSoA };
+export { writeSoA };

package/core/buffer/buffer.d.ts

@@ -3,8 +3,11 @@ import { $internal } from "../../shared/symbols.js";
 import { Prettify, UnionToIntersection } from "../../shared/utilityTypes.js";
 import { StorageFlag } from "../../extension.js";
 import { TgpuBufferMutable, TgpuBufferReadonly, TgpuBufferUniform, TgpuFixedBufferUsage } from "./bufferUsage.js";
+import "../root/rootTypes.js";
 import { BaseData } from "../../data/wgslTypes.js";
 import { Infer, InferInput, InferPartial, InferPatch, IsValidIndexSchema, IsValidStorageSchema, IsValidUniformSchema, IsValidVertexSchema, MemIdentity } from "../../shared/repr.js";
+import "../../data/dataTypes.js";
+
 //#region src/core/buffer/buffer.d.ts
 interface UniformFlag {
   usableAsUniform: true;

package/core/buffer/buffer.js

@@ -10,6 +10,7 @@ import { readData, writeData } from "../../data/dataIO.js";
 import { convertPartialToPatch, getPatchInstructions } from "../../data/partialIO.js";
 import { mutable, readonly, uniform } from "./bufferUsage.js";
 import { BufferReader, BufferWriter, getSystemEndianness } from "typed-binary";
+
 //#region src/core/buffer/buffer.ts
 const usageToUsageConstructor = {
 	uniform,
@@ -212,9 +213,9 @@ var TgpuBufferImpl = class {
 		if (gpuBuffer.mapState === "mapped") return readData(new BufferReader(this.#getMappedRange()), this.dataType);
 		if (gpuBuffer.usage & GPUBufferUsage.MAP_READ) {
 			await gpuBuffer.mapAsync(GPUMapMode.READ);
-			const res = readData(new BufferReader(this.#getMappedRange()), this.dataType);
+			const res$1 = readData(new BufferReader(this.#getMappedRange()), this.dataType);
 			this.#unmapBuffer();
-			return res;
+			return res$1;
 		}
 		const stagingBuffer = this.#device.createBuffer({
 			size: sizeOf(this.dataType),
@@ -242,5 +243,6 @@ var TgpuBufferImpl = class {
 		return `buffer:${getName(this) ?? "<unnamed>"}`;
 	}
 };
+
 //#endregion
-export { INTERNAL_createBuffer, isBuffer, isUsableAsVertex };
+export { INTERNAL_createBuffer, isBuffer, isUsableAsVertex };

package/core/buffer/bufferShorthand.d.ts

@@ -1,7 +1,9 @@
 import { TgpuNamable } from "../../shared/meta.js";
 import { $gpuValueOf, $internal } from "../../shared/symbols.js";
 import { StorageFlag } from "../../extension.js";
+import "../../data/snippet.js";
 import { BufferWriteOptions, TgpuBuffer, UniformFlag } from "./buffer.js";
+import "../../types.js";
 import { BaseData } from "../../data/wgslTypes.js";
 import { Infer, InferGPU, InferInput, InferPartial, InferPatch } from "../../shared/repr.js";
 

package/core/buffer/bufferShorthand.js

@@ -1,5 +1,6 @@
 import { $getNameForward, $gpuValueOf, $internal, $resolve } from "../../shared/symbols.js";
 import { getName, setName } from "../../shared/meta.js";
+
 //#region src/core/buffer/bufferShorthand.ts
 function isBufferShorthand(value) {
 	return value instanceof TgpuBufferShorthandImpl;
@@ -49,5 +50,6 @@ var TgpuBufferShorthandImpl = class {
 		return ctx.resolve(this.#usage);
 	}
 };
+
 //#endregion
-export { TgpuBufferShorthandImpl, isBufferShorthand };
+export { TgpuBufferShorthandImpl, isBufferShorthand };

package/core/buffer/bufferUsage.d.ts

@@ -1,5 +1,7 @@
 import { TgpuNamable } from "../../shared/meta.js";
 import { $gpuValueOf, $internal, $repr } from "../../shared/symbols.js";
+import "../../data/snippet.js";
+import "../../tgpuBindGroupLayout.js";
 import { TgpuBuffer, UniformFlag } from "./buffer.js";
 import { BindableBufferUsage } from "../../types.js";
 import { BaseData } from "../../data/wgslTypes.js";

package/core/buffer/bufferUsage.js

@@ -8,6 +8,7 @@ import { schemaCallWrapper } from "../../data/schemaCallWrapper.js";
 import { assertExhaustive } from "../../shared/utilityTypes.js";
 import { valueProxyHandler } from "../valueProxyUtils.js";
 import { isUsableAsStorage } from "../../extension.js";
+
 //#region src/core/buffer/bufferUsage.ts
 function isUsableAsUniform(buffer) {
 	return !!buffer.usableAsUniform;
@@ -161,5 +162,6 @@ function uniform(buffer) {
 	}
 	return usage;
 }
+
 //#endregion
-export { TgpuLaidOutBufferImpl, isUsableAsUniform, mutable, readonly, uniform };
+export { TgpuLaidOutBufferImpl, isUsableAsUniform, mutable, readonly, uniform };

package/core/constant/tgpuConstant.d.ts

@@ -17,7 +17,8 @@ interface TgpuConst<TDataType extends BaseData = BaseData> extends TgpuNamable {
   readonly value: DeepReadonly<InferGPU<TDataType>>;
   readonly $: DeepReadonly<InferGPU<TDataType>>;
   readonly [$internal]: {
-    /** Makes it differentiable on the type level. Does not exist at runtime. */dataType?: TDataType;
+    /** Makes it differentiable on the type level. Does not exist at runtime. */
+    dataType?: TDataType;
   };
 }
 /**

package/core/constant/tgpuConstant.js

@@ -5,6 +5,7 @@ import { isData } from "../../data/dataTypes.js";
 import { snip } from "../../data/snippet.js";
 import { inCodegenMode } from "../../execMode.js";
 import { valueProxyHandler } from "../valueProxyUtils.js";
+
 //#region src/core/constant/tgpuConstant.ts
 function constant(dataType, value) {
 	if (!isData(dataType)) {
@@ -64,5 +65,6 @@ var TgpuConstImpl = class {
 		return this.$;
 	}
 };
+
 //#endregion
-export { constant };
+export { constant };

package/core/declare/tgpuDeclare.js

@@ -2,6 +2,7 @@ import { $internal, $resolve } from "../../shared/symbols.js";
 import { Void } from "../../data/wgslTypes.js";
 import { snip } from "../../data/snippet.js";
 import { applyExternals, replaceExternalsInWgsl } from "../resolve/externals.js";
+
 //#region src/core/declare/tgpuDeclare.ts
 /**
 * Allows defining extra declarations that shall be included in the final WGSL code,
@@ -35,5 +36,6 @@ var TgpuDeclareImpl = class {
 		return `declare: ${this.#declaration}`;
 	}
 };
+
 //#endregion
-export { declare };
+export { declare };

package/core/function/autoIO.d.ts

@@ -1,8 +1,11 @@
 import { Assume, RemoveIndexSignature } from "../../shared/utilityTypes.js";
 import { BuiltinClipDistances, BuiltinFragDepth, BuiltinFrontFacing, BuiltinInstanceIndex, BuiltinPosition, BuiltinPrimitiveIndex, BuiltinSampleIndex, BuiltinSampleMask, BuiltinSubgroupInvocationId, BuiltinSubgroupSize, BuiltinVertexIndex, OmitBuiltins } from "../../builtin.js";
+import "../../data/snippet.js";
 import { BaseIOData } from "./fnTypes.js";
 import { TgpuVertexAttrib } from "../../shared/vertexFormat.js";
+import "../../data/autoStruct.js";
 import { FormatToWGSLType } from "../../data/vertexFormatData.js";
+import "../../types.js";
 import { v4f } from "../../data/wgslTypes.js";
 import { InferGPU, InferGPURecord, InferRecord } from "../../shared/repr.js";
 

package/core/function/autoIO.js

@@ -5,6 +5,7 @@ import { AutoStruct } from "../../data/autoStruct.js";
 import { createFnCore } from "./fnCore.js";
 import { shaderStageSlot } from "../slot/internalSlots.js";
 import { builtin } from "../../builtin.js";
+
 //#region src/core/function/autoIO.ts
 const builtinVertexIn = {
 	$vertexIndex: builtin.vertexIndex,
@@ -36,7 +37,7 @@ var AutoFragmentFn = class {
 	autoOut;
 	constructor(impl, varyings, locations) {
 		if (!getName(impl)) setName(impl, "fragmentFn");
-		this.#core = createFnCore(impl, "@fragment ");
+		this.#core = createFnCore(impl, "fragment");
 		this.autoIn = new AutoStruct({
 			...builtinFragmentIn,
 			...varyings
@@ -63,7 +64,7 @@ var AutoVertexFn = class {
 	autoOut;
 	constructor(impl, attribs, locations) {
 		if (!getName(impl)) setName(impl, "vertexFn");
-		this.#core = createFnCore(impl, "@vertex ");
+		this.#core = createFnCore(impl, "vertex");
 		this.autoIn = new AutoStruct({
 			...builtinVertexIn,
 			...attribs
@@ -81,5 +82,6 @@ var AutoVertexFn = class {
 };
 AutoVertexFn.prototype[$internal] = true;
 AutoVertexFn.prototype.resourceType = "auto-vertex-fn";
+
 //#endregion
-export { AutoFragmentFn, AutoVertexFn };
+export { AutoFragmentFn, AutoVertexFn };

package/core/function/comptime.js

@@ -3,6 +3,7 @@ import { setName } from "../../shared/meta.js";
 import { WgslTypeError } from "../../errors.js";
 import { isKnownAtComptime } from "../../types.js";
 import { coerceToSnippet } from "../../tgsl/generationHelpers.js";
+
 //#region src/core/function/comptime.ts
 /**
 * Creates a version of `func` that can called safely in a TypeGPU function to
@@ -45,5 +46,6 @@ function comptime(func) {
 	Object.defineProperty(impl, $internal, { value: { isComptime: true } });
 	return impl;
 }
+
 //#endregion
-export { comptime };
+export { comptime };

package/core/function/createCallableSchema.js

@@ -4,6 +4,7 @@ import { isPtr } from "../../data/wgslTypes.js";
 import { snip } from "../../data/snippet.js";
 import { NormalState, isKnownAtComptime } from "../../types.js";
 import { tryConvertSnippet } from "../../tgsl/conversion.js";
+
 //#region src/core/function/createCallableSchema.ts
 function callableSchema(options) {
 	const impl = ((...args) => {
@@ -36,5 +37,6 @@ function callableSchema(options) {
 	};
 	return impl;
 }
+
 //#endregion
-export { callableSchema };
+export { callableSchema };

package/core/function/dualImpl.js

@@ -5,6 +5,7 @@ import { snip } from "../../data/snippet.js";
 import { NormalState, isKnownAtComptime } from "../../types.js";
 import { tryConvertSnippet } from "../../tgsl/conversion.js";
 import { concretize } from "../../tgsl/generationHelpers.js";
+
 //#region src/core/function/dualImpl.ts
 var MissingCpuImplError = class extends Error {
 	constructor(message) {
@@ -48,5 +49,6 @@ function dualImpl(options) {
 	};
 	return impl;
 }
+
 //#endregion
-export { MissingCpuImplError, dualImpl };
+export { MissingCpuImplError, dualImpl };

package/core/function/entryInputRouter.js

@@ -1,7 +1,6 @@
 import { $internal } from "../../shared/symbols.js";
 import { isWgslStruct } from "../../data/wgslTypes.js";
-import { undecorate } from "../../data/dataTypes.js";
-import { snip } from "../../data/snippet.js";
+
 //#region src/core/function/entryInputRouter.ts
 /**
 * Routes `(input) => { input.x }` style property access to the correct WGSL
@@ -11,29 +10,26 @@ import { snip } from "../../data/snippet.js";
 var EntryInputRouter = class {
 	[$internal] = {};
 	type = "entry-input-router";
-	structArgName;
-	dataSchema;
+	structArg;
 	/** Maps schemaKey → { WGSL arg name, type } */
 	positionalArgsMap;
-	constructor(structArgName, dataSchema, positionalArgs) {
-		this.structArgName = structArgName;
-		this.dataSchema = dataSchema;
-		this.positionalArgsMap = new Map(positionalArgs.map((a) => [a.schemaKey, {
-			argName: a.argName,
-			type: a.type
-		}]));
+	constructor(structArg, positionalArgs) {
+		this.structArg = structArg;
+		this.positionalArgsMap = new Map(positionalArgs.map((a) => [a.schemaKey, a.arg]));
 	}
 	toString() {
 		return "entry-input-router";
 	}
 	accessProp(propName) {
 		const positionalEntry = this.positionalArgsMap.get(propName);
-		if (positionalEntry) return snip(positionalEntry.argName, positionalEntry.type, "argument");
-		if (this.dataSchema && isWgslStruct(this.dataSchema)) {
-			const propType = this.dataSchema.propTypes[propName];
-			if (propType) return snip(`${this.structArgName}.${propName}`, undecorate(propType), "argument");
-		}
+		if (positionalEntry) return positionalEntry();
+		const structSnippet = this.structArg?.();
+		if (structSnippet && isWgslStruct(structSnippet.dataType)) return {
+			target: structSnippet,
+			prop: propName
+		};
 	}
 };
+
 //#endregion
-export { EntryInputRouter };
+export { EntryInputRouter };

package/core/function/extractArgs.js

@@ -200,5 +200,6 @@ const openingCommentBlock = new Set(["*/"]);
 const parentheses = ["(", ")"];
 const angleBrackets = ["<", ">"];
 const commentBlocks = ["/*", "*/"];
+
 //#endregion
-export { extractArgs };
+export { extractArgs };

package/core/function/fnCore.js

@@ -7,8 +7,9 @@ import { MissingLinksError } from "../../errors.js";
 import { getAttributesString } from "../../data/attributes.js";
 import { applyExternals, replaceExternalsInWgsl } from "../resolve/externals.js";
 import { extractArgs } from "./extractArgs.js";
+
 //#region src/core/function/fnCore.ts
-function createFnCore(implementation, fnAttribute = "") {
+function createFnCore(implementation, functionType, workgroupSize) {
 	/**
 	* External application has to be deferred until resolution because
 	* some externals can reference the owner function which has not been
@@ -23,6 +24,10 @@ function createFnCore(implementation, fnAttribute = "") {
 		},
 		resolve(ctx, argTypes, returnType, entryInput) {
 			const externalMap = {};
+			let attributes = "";
+			if (functionType === "compute") attributes = `@compute @workgroup_size(${workgroupSize?.join(", ")}) `;
+			else if (functionType === "vertex") attributes = `@vertex `;
+			else if (functionType === "fragment") attributes = `@fragment `;
 			for (const externals of externalsToApply) applyExternals(externalMap, externals);
 			const id = ctx.getUniqueName(this);
 			if (typeof implementation === "string") {
@@ -32,7 +37,7 @@ function createFnCore(implementation, fnAttribute = "") {
 				const replacedImpl = replaceExternalsInWgsl(ctx, externalMap, implementation);
 				let header = "";
 				let body = "";
-				if (fnAttribute !== "" && entryInput && validArgNames) {
+				if (functionType !== "normal" && entryInput && validArgNames) {
 					const { dataSchema, positionalArgs } = entryInput;
 					const parts = [];
 					if (dataSchema && isArgUsedInBody("in", replacedImpl)) parts.push(`in: ${ctx.resolve(dataSchema).value}`);
@@ -41,8 +46,8 @@ function createFnCore(implementation, fnAttribute = "") {
 						if (argName !== "" && isArgUsedInBody(argName, replacedImpl)) parts.push(`${getAttributesString(a.type)}${argName}: ${ctx.resolve(a.type).value}`);
 					}
 					const input = `(${parts.join(", ")})`;
-					const attributes = isWgslData(returnType) ? getAttributesString(returnType) : "";
-					header = `${input}${returnType !== Void ? isWgslStruct(returnType) ? ` -> ${ctx.resolve(returnType).value} ` : ` -> ${attributes !== "" ? attributes : "@location(0)"} ${ctx.resolve(returnType).value} ` : " "}`;
+					const attributes$1 = isWgslData(returnType) ? getAttributesString(returnType) : "";
+					header = `${input}${returnType !== Void ? isWgslStruct(returnType) ? ` -> ${ctx.resolve(returnType).value} ` : ` -> ${attributes$1 !== "" ? attributes$1 : "@location(0)"} ${ctx.resolve(returnType).value} ` : " "}`;
 					body = replacedImpl;
 				} else {
 					const providedArgs = extractArgs(replacedImpl);
@@ -50,7 +55,7 @@ function createFnCore(implementation, fnAttribute = "") {
 					header = `(${providedArgs.args.map((argInfo, i) => `${argInfo.identifier}: ${checkAndReturnType(ctx, `parameter ${argInfo.identifier}`, argInfo.type, argTypes[i])}`).join(", ")})${returnType === Void ? " " : ` -> ${checkAndReturnType(ctx, "return type", providedArgs.ret?.type, returnType)} `}`;
 					body = replacedImpl.slice(providedArgs.range.end);
 				}
-				ctx.addDeclaration(`${fnAttribute}fn ${id}${header}${body}`);
+				ctx.addDeclaration(`${attributes}fn ${id}${header}${body}`);
 				return snip(id, returnType, "runtime");
 			}
 			const pluginData = getMetaData(implementation);
@@ -61,9 +66,9 @@ function createFnCore(implementation, fnAttribute = "") {
 			const missingExternals = ast.externalNames.filter((name) => !(name in externalMap));
 			if (missingExternals.length > 0) throw new MissingLinksError(getName(this), missingExternals);
 			const maybeSecondArg = ast.params[1];
-			if (maybeSecondArg && maybeSecondArg.type === "i" && fnAttribute !== "") applyExternals(externalMap, { [maybeSecondArg.name]: undecorate(returnType) });
-			const { head, body, returnType: actualReturnType } = ctx.fnToWgsl({
-				functionType: fnAttribute.includes("@compute") ? "compute" : fnAttribute.includes("@vertex") ? "vertex" : fnAttribute.includes("@fragment") ? "fragment" : "normal",
+			if (maybeSecondArg && maybeSecondArg.type === "i" && functionType !== "normal") applyExternals(externalMap, { [maybeSecondArg.name]: undecorate(returnType) });
+			const { code, returnType: actualReturnType } = ctx.fnToWgsl({
+				functionType,
 				argTypes,
 				entryInput,
 				params: ast.params,
@@ -71,13 +76,13 @@ function createFnCore(implementation, fnAttribute = "") {
 				body: ast.body,
 				externalMap
 			});
-			ctx.addDeclaration(`${fnAttribute}fn ${id}${ctx.resolve(head).value}${ctx.resolve(body).value}`);
+			ctx.addDeclaration(`${attributes}fn ${id}${code}`);
 			return snip(id, actualReturnType, "runtime");
 		}
 	};
 }
 function isArgUsedInBody(argName, body) {
-	return new RegExp(`\\b${argName}\\b`).test(body);
+	return (/* @__PURE__ */ new RegExp(`\\b${argName}\\b`)).test(body);
 }
 function checkAndReturnType(ctx, name, wgslType, jsType) {
 	const resolvedJsType = ctx.resolve(jsType).value.replace(/\s/g, "");
@@ -86,5 +91,6 @@ function checkAndReturnType(ctx, name, wgslType, jsType) {
 	if (resolvedWgslType !== resolvedJsType) throw new Error(`Type mismatch between TGPU shell and WGSL code string: ${name}, JS type "${resolvedJsType}", WGSL type "${resolvedWgslType}".`);
 	return wgslType;
 }
+
 //#endregion
-export { createFnCore };
+export { createFnCore };

package/core/function/fnTypes.d.ts

@@ -2,6 +2,8 @@ import { BuiltinClipDistances } from "../../builtin.js";
 import { AnyAttribute } from "../../data/attributes.js";
 import { BaseData, Bool, Decorated, F16, F32, I32, U32, Vec2f, Vec2h, Vec2i, Vec2u, Vec3f, Vec3h, Vec3i, Vec3u, Vec4f, Vec4h, Vec4i, Vec4u, Void } from "../../data/wgslTypes.js";
 import { InferGPU } from "../../shared/repr.js";
+import "tinyest";
+
 //#region src/core/function/fnTypes.d.ts
 type AnyFn = (...args: never[]) => unknown;
 type InferArgs<T extends unknown[]> = { [Idx in keyof T]: InferGPU<T[Idx]> };

package/core/function/ioSchema.d.ts

@@ -1,4 +1,5 @@
 import { Decorate, HasCustomLocation, IsBuiltin } from "../../data/attributes.js";
+import "./fnTypes.js";
 import { BaseData, Location, WgslStruct } from "../../data/wgslTypes.js";
 
 //#region src/core/function/ioSchema.d.ts

package/core/function/ioSchema.js

@@ -2,6 +2,7 @@ import { isVoid } from "../../data/wgslTypes.js";
 import { getCustomLocation, isData } from "../../data/dataTypes.js";
 import { isBuiltin, location } from "../../data/attributes.js";
 import { INTERNAL_createStruct } from "../../data/struct.js";
+
 //#region src/core/function/ioSchema.ts
 function withLocations(members, locations = {}) {
 	let nextLocation = 0;
@@ -47,5 +48,6 @@ function separateAllAsPositional(schema) {
 function createIoSchema(layout, locations = {}) {
 	return isData(layout) ? isVoid(layout) ? layout : isBuiltin(layout) ? layout : getCustomLocation(layout) !== void 0 ? layout : location(0, layout) : INTERNAL_createStruct(withLocations(layout, locations), false);
 }
+
 //#endregion
-export { createIoSchema, separateAllAsPositional, separateBuiltins };
+export { createIoSchema, separateAllAsPositional, separateBuiltins };

package/core/function/shelllessImpl.d.ts

@@ -3,6 +3,7 @@ import { SelfResolvable } from "../../types.js";
 import { BaseData } from "../../data/wgslTypes.js";
 
 //#region src/core/function/shelllessImpl.d.ts
+
 /**
  * Shell-less functions are possible because we can infer the signature based solely on the context
  * around the function.

package/core/function/shelllessImpl.js

@@ -1,9 +1,10 @@
 import { $getNameForward, $internal, $resolve } from "../../shared/symbols.js";
 import { getName } from "../../shared/meta.js";
 import { createFnCore } from "./fnCore.js";
+
 //#region src/core/function/shelllessImpl.ts
 function createShelllessImpl(argTypes, implementation) {
-	const core = createFnCore(implementation, "");
+	const core = createFnCore(implementation, "normal");
 	return {
 		[$internal]: true,
 		[$getNameForward]: core,
@@ -17,5 +18,6 @@ function createShelllessImpl(argTypes, implementation) {
 		}
 	};
 }
+
 //#endregion
-export { createShelllessImpl };
+export { createShelllessImpl };

package/core/function/templateUtils.js

@@ -8,5 +8,6 @@ function isTemplateStringsArray(value) {
 function templateLiteralIdentity(strings, ...values) {
 	return strings.slice(1).reduce((acc, elem, index) => `${acc}${values[index]}${elem}`, strings[0]);
 }
+
 //#endregion
-export { stripTemplate };
+export { stripTemplate };

package/core/function/tgpuComputeFn.d.ts

@@ -5,6 +5,7 @@ import { IORecord, InferIO } from "./fnTypes.js";
 import { Void } from "../../data/wgslTypes.js";
 
 //#region src/core/function/tgpuComputeFn.d.ts
+
 /**
  * Describes a compute entry function signature (its arguments, return type and workgroup size)
  */

package/core/function/tgpuComputeFn.js

@@ -5,6 +5,7 @@ import { separateBuiltins } from "./ioSchema.js";
 import { createFnCore } from "./fnCore.js";
 import { stripTemplate } from "./templateUtils.js";
 import { shaderStageSlot } from "../slot/internalSlots.js";
+
 //#region src/core/function/tgpuComputeFn.ts
 /**
 * Creates a shell of a typed entry function for the compute shader stage. Any function
@@ -30,7 +31,7 @@ function isTgpuComputeFn(value) {
 	return value?.shell?.entryPoint === "compute";
 }
 function createComputeFn(shell, workgroupSize, implementation, entryInput) {
-	const core = createFnCore(implementation, `@compute @workgroup_size(${workgroupSize.join(", ")}) `);
+	const core = createFnCore(implementation, "compute", workgroupSize);
 	return {
 		shell,
 		$uses(newExternals) {
@@ -51,5 +52,6 @@ function createComputeFn(shell, workgroupSize, implementation, entryInput) {
 		}
 	};
 }
+
 //#endregion
-export { computeFn, isTgpuComputeFn };
+export { computeFn, isTgpuComputeFn };

package/core/function/tgpuFn.d.ts

@@ -9,6 +9,7 @@ import { BaseData, Void } from "../../data/wgslTypes.js";
 import { InferGPU } from "../../shared/repr.js";
 
 //#region src/core/function/tgpuFn.d.ts
+
 /**
  * Describes a function signature (its arguments and return type)
  */