Rust-GPU/rust-cuda
Rust Cuda
rust-cuda est un framework de programmation GPU et un compilateur de périphériques qui permet le développement et l'exécution de noyaux (kernels) haute performance sur du matériel NVIDIA en utilisant Rust. Il fournit un wrapper de pilote pour gérer l'allocation de mémoire du périphérique et le lancement des noyaux, servant efficacement de système pour écrire une logique de calcul GPU sans dépendre du C++.
Le projet inclut une bibliothèque de calcul avec des primitives optimisées pour le matériel pour l'accélération des réseaux neuronaux et le raytracing accéléré par le matériel. Il utilise une chaîne de compilation qui traduit le code source en une représentation intermédiaire de bas niveau pour l'exécution sur les processeurs graphiques.
Le framework couvre la gestion des ressources du périphérique, le développement de noyaux et la simulation d'opérations entières de haute précision. Il prend également en charge la génération de nombres aléatoires côté périphérique et des optimisations de calcul spécifiques à la cible.
Des images de conteneurs pré-configurées sont disponibles pour simplifier le provisionnement de la chaîne de compilation et de l'environnement de développement à travers différentes architectures matérielles.
Features
- GPU Compilers - Provides a complete toolchain that translates Rust source code into optimized NVIDIA GPU hardware instructions.
- GPU Kernel Programming - Provides a framework for writing and executing high-performance GPU kernels using Rust.
- Compute Libraries - Ships a compute library with hardware-optimized primitives for neural networks and raytracing.
- GPU Resource Management - Implements a high-level driver interface to manage the lifecycle and allocation of GPU buffers and kernels.
- CUDA Driver Wrappers - Provides a programmatic abstraction for controlling NVIDIA GPU device memory and kernel execution.
- GPU Kernel Development - Provides a framework for managing thread indices and memory allocation to create device-side logic.
- CUDA Kernel Compilers - Compiles GPU source code into target-specific CUDA kernels with a Rust interface.
- GPU Compute Toolchains - Provides an integrated development environment and compiler for writing high-performance Rust compute kernels.
- Intermediate Representations - Compiles Rust source code into a PTX-based intermediate representation for GPU execution.
- CUDA Accelerated Neural Networks - Implements hardware-optimized primitives to accelerate deep learning workloads on NVIDIA GPUs.
- GPU Kernel Primitives - Provides highly tuned GPU kernels for standard deep learning operations tuned for tensor cores.
- Raytracing Primitives - Interfaces directly with specialized GPU circuits to compute light paths and perform image denoising.
- Raytracing Simulations - Uses specialized GPU circuits to implement physical light transport and path tracing simulations.
- Real-Time Raytracing - Implements real-time raytracing techniques using specialized hardware circuits for physical light transport.
- Hardware Driver API Mappings - Provides mappings that link high-level programmatic calls to native NVIDIA hardware driver and runtime APIs.
- Binary Instruction Execution - Executes low-level binary instructions on graphics hardware using a specialized intermediate representation.
- Compilation Target Specifications - Defines specific hardware compute capabilities during compilation to optimize machine code for target versions.
- Hardware-Targeted Compilation - Configures compilation rules to optimize binaries for specific GPU hardware revisions and architectures.
