fix: Better type exports

gpujs · Jan 13, 2021 · 026763a · 026763a
1 parent 7e3fa3a
commit 026763a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "gpu.js",
-  "version": "2.11.0",
+  "version": "2.11.1",
   "description": "GPU Accelerated JavaScript",
   "engines": {
     "node": ">=8.0.0"

diff --git a/src/index.d.ts b/src/index.d.ts
@@ -13,7 +13,7 @@ export class GPU {
   nativeFunctions: IGPUNativeFunction[];
   setFunctions(flag: any): this;
   setNativeFunctions(flag: IGPUNativeFunction[]): this;
-  addFunction(kernel: GPUFunction, settings?: IGPUFunctionSettings): this;
+  addFunction<ArgTypes extends ThreadKernelVariable[] = ThreadKernelVariable[], ConstantsType = {}>(kernel: GPUFunction<ArgTypes, ConstantsType>, settings?: IGPUFunctionSettings): this;
   addNativeFunction(name: string, source: string, settings?: IGPUFunctionSettings): this;
   combineKernels(...kernels: KernelFunction[]): IKernelRunShortcut;
   combineKernels<KF extends KernelFunction>(...kernels: KF[]):
@@ -617,7 +617,7 @@ export class Input {
 
 export type input = (value: number[], size: OutputDimensions) => Input;
 
-export function alias(name: string, source: KernelFunction): KernelFunction;
+export function alias<T>(name: string, source: T): T;
 
 export class KernelValue {
   constructor(value: KernelVariable, settings: IKernelValueSettings);
@@ -657,8 +657,8 @@ export interface IJSONSettings {
   threadDim?: number[];
 }
 
-export type utils = {
-  getMinifySafeName(arrowReference: () => Function): string;
+export declare const utils: {
+  getMinifySafeName: <T>(arrowReference: () => T) => string
 }
 
 export interface IReason {