[package] name = "rtx-cubecl" version.workspace = true edition.workspace = true authors.workspace = true license.workspace = true repository.workspace = true description = "CubeCL-based GPU kernel compilation for RustyTorch++ - write GPU kernels in Rust" [dependencies] # CubeCL core - Rust to GPU kernel compilation cubecl.workspace = true cubecl-core.workspace = true cubecl-runtime.workspace = true # Backend-specific CubeCL runtimes cubecl-wgpu = { workspace = true, optional = true } cubecl-cuda = { workspace = true, optional = true } cubecl-hip = { workspace = true, optional = true } # Internal RTX crates rtx-backend = { workspace = true } # Error handling thiserror.workspace = true anyhow.workspace = true # Logging tracing.workspace = true # Async support async-trait.workspace = true [dev-dependencies] tokio-test.workspace = true [features] default = [] # CPU backend (uses CubeCL's CPU runtime when available) cpu = [] # WebGPU backend - cross-platform GPU (browsers, desktop) wgpu = ["dep:cubecl-wgpu"] # CUDA backend - NVIDIA GPUs cuda = ["dep:cubecl-cuda"] # HIP/ROCm backend - AMD GPUs hip = ["dep:cubecl-hip"] # Vulkan backend - via WebGPU/SPIR-V vulkan = ["wgpu"] # Metal backend - Apple GPUs via WebGPU metal = ["wgpu"] # All GPU backends full = ["wgpu", "cuda", "hip"] [lints] workspace = true