software-mansion/TypeGPU
TypeGPU
TypeGPU is a tool for type-safe WebGPU development that enables writing shaders in TypeScript. It translates high-level TypeScript function definitions and structures into WebGPU Shading Language source code to automate shader generation and validate logic using a type system.
The project provides a mechanism for cross-library GPU interoperability by sharing typed buffers without copying data to system memory. It also integrates the Model Context Protocol to allow AI agents to inspect generated shader code and diagnose runtime errors.
The system manages WebGPU resource mapping through typed layouts for bind groups and uses schema-based binary encoding to handle memory alignment and buffer sizes. It further supports the definition of GPU data types to ensure consistent data transfer between the processor and the graphics unit.
Features
- Type-Safe Shader Generation - Translates high-level TypeScript function definitions into WebGPU Shading Language source code to automate shader generation.
- TypeScript-to-WGSL Compilers - Translates high-level TypeScript function definitions into WebGPU Shading Language source code.
- GPU Memory Alignment Encoders - Translates complex data structures into typed binary formats to ensure correct memory alignment during CPU-to-GPU transfer.
- GPU Buffer Allocators - Enforces typed read/write operations by calculating memory sizes and binary padding for GPU buffers.
- User-Defined Types - Defines custom GPU data structures and arrays to ensure data matches the expected schema.
- Schema-Based Binary Formats - Converts complex data structures into typed binary formats using defined layouts to ensure correct GPU memory alignment.
- Zero-Copy GPU Buffer Interop - Shares typed GPU buffers between libraries without copying data to system memory for high performance.
- Native Type Mappings - Links raw hardware resources to typed interfaces to retrieve and manage underlying native handles.
- GPU Bind Group Mappings - Provides automatic mapping of resource keys to GPU binding indices and validates resource types during pipeline setup.
- GPU Resource Management - Manages the mapping of buffers and bind groups between TypeScript and the GPU to ensure data alignment and consistency.
- Function-to-Shader Transpilers - Translates typed TypeScript functions into valid WGSL strings to execute logic on the GPU.
- Type-Safe Shader Generation - Transforms typed functions into shader code to validate logic using a high-level type system.
- Shader Resource Binders - Produces shader code from typed structures to keep resource definitions consistent via defined layouts.
- Typed Bind Group Management - Maps hardware resources to named keys to handle binding indices and validate resource types automatically.
- GPU Buffer Sharing Allocations - Passes typed GPU buffers between libraries without copying data to system memory for interoperability.
- GPU Memory Layout Definitions - Calculates buffer sizes and applies binary padding based on TypeScript definitions to meet strict GPU memory requirements.
- Resource Handle Mapping - Links native GPU resource handles to typed API wrappers to maintain type safety during resource management.
- Data Encoders and Decoders - Translates complex TypeScript data structures into typed binary formats for efficient CPU-to-GPU transfer.
- Model Context Protocol Integrations - Implements the Model Context Protocol to expose generated shader code and diagnostics to AI agents.
- AI-Powered Shader Debugging - Uses the Model Context Protocol to let AI agents inspect generated shader code and diagnose GPU runtime errors.
- AI Assistant Integrations - Connects AI coding assistants to the development environment to inspect generated shader code.
- Zero-Copy Buffer Interoperability - Provides a mechanism for sharing typed GPU buffers between libraries without copying data to system memory.
