56 lines
1.3 KiB
TOML
56 lines
1.3 KiB
TOML
[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
|