fix: Use InferGPU in function shell types (#2295)

Author: iwoplaza

packages/typegpu/src/core/function/fnTypes.ts

@@ -23,7 +23,7 @@ import type {
   Vec4u,
   Void,
 } from '../../data/wgslTypes.ts';
-import type { Infer } from '../../shared/repr.ts';
+import type { InferGPU } from '../../shared/repr.ts';
 
 export type AnyFn = (...args: never[]) => unknown;
 
@@ -41,7 +41,7 @@ export type TranspilationResult = {
 };
 
 export type InferArgs<T extends unknown[]> = {
-  [Idx in keyof T]: Infer<T[Idx]>;
+  [Idx in keyof T]: InferGPU<T[Idx]>;
 };
 
 type InheritTupleValues<T, From> = {
@@ -65,7 +65,7 @@ export type InheritArgNames<T extends AnyFn, From extends AnyFn> = {
 
 export type InferImplSchema<ImplSchema extends AnyFn> = (
   ...args: InferArgs<Parameters<ImplSchema>>
-) => Infer<ReturnType<ImplSchema>>;
+) => InferGPU<ReturnType<ImplSchema>>;
 
 export type Implementation<ImplSchema extends AnyFn = AnyFn> = string | InferImplSchema<ImplSchema>;
 
@@ -101,9 +101,9 @@ export type IOLayout<TElementType extends IOData = IOData> =
   | Void;
 
 export type InferIO<T> = T extends { type: string }
-  ? Infer<T>
+  ? InferGPU<T>
   : T extends Record<string, unknown>
-    ? { [K in keyof T]: Infer<T[K]> }
+    ? { [K in keyof T]: InferGPU<T[K]> }
     : T;
 
 export interface PositionalArgInfo {

packages/typegpu/src/core/function/tgpuFn.ts

@@ -6,7 +6,7 @@ import { provideInsideTgpuFn } from '../../execMode.ts';
 import type { TgpuNamable } from '../../shared/meta.ts';
 import { getName, setName } from '../../shared/meta.ts';
 import { isMarkedInternal } from '../../shared/symbols.ts';
-import type { Infer } from '../../shared/repr.ts';
+import type { InferGPU } from '../../shared/repr.ts';
 import { $getNameForward, $internal, $providing, $resolve } from '../../shared/symbols.ts';
 import type { Prettify } from '../../shared/utilityTypes.ts';
 import type { DualFn, ResolutionCtx, SelfResolvable } from '../../types.ts';
@@ -56,7 +56,7 @@ export type TgpuFnShell<Args extends BaseData[], Return extends BaseData> = Tgpu
   Args,
   Return
 > &
-  (<T extends (...args: InferArgs<Args>) => Infer<Return>>(
+  (<T extends (...args: InferArgs<Args>) => InferGPU<Return>>(
     implementation: T,
   ) => TgpuFn<Prettify<InheritArgNames<(...args: Args) => Return, T>>['result']>) &
   ((implementation: string) => TgpuFn<(...args: Args) => Return>) &