feat: tgpufn callback

[lursz]

A new overload for tgpu.fn that accepts a shell-less callback and returns a callable that has a .with() method. This allows shell-less functions to also be provided with slot and accessor overrides.

Example

const countAccess = tgpu.accessor(d.f32);

const getDouble = () => {
  'use gpu';
  return countAccess.$ * 2;
};
// ^? () => number

const getDouble4 = tgpu.fn(getDouble).with(countAccess, 4);
//    ^? TgpuGenericFn<() => number>

const main = () => {
  'use gpu';
  return getDouble4(); // should return 8
};

Requirements

• If a shell-less function foo is used in code, and tgpu.fn(foo) is also used in code, it should only generate one definition.

[github-actions]

pkg.pr.new

packages
Ready to be installed by your favorite package manager ⬇️

https://pkg.pr.new/software-mansion/TypeGPU/typegpu@5a92b036a116220e0d28552ddc9c81a64db66de1

https://pkg.pr.new/software-mansion/TypeGPU/@typegpu/noise@5a92b036a116220e0d28552ddc9c81a64db66de1

https://pkg.pr.new/software-mansion/TypeGPU/unplugin-typegpu@5a92b036a116220e0d28552ddc9c81a64db66de1

benchmark
view benchmark

commit
view commit