rustyverse/rustytorch
2
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

D310: enable rtx-backend-cuda for Blackwell sm_120 / CUDA 13.1
Performance Benchmarks / Run Benchmarks (push) Has been cancelled
Details
Documentation / Build API Documentation (push) Has been cancelled
Details
Documentation / Build User Guide (push) Has been cancelled
Details
CI / Format Check (push) Has been cancelled
Details
CI / Clippy Check (push) Has been cancelled
Details
CI / Build (macos-latest) (push) Has been cancelled
Details
CI / Build (ubuntu-latest) (push) Has been cancelled
Details
CI / Build CPU-Only (Explicit) (push) Has been cancelled
Details
CI / Test (macos-latest) (push) Has been cancelled
Details
CI / Test (ubuntu-latest) (push) Has been cancelled
Details
CI / CI Success (push) Has been cancelled
Details

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <[email protected]>
This commit is contained in:
Omar Sobh
2026-06-22 12:02:37 +00:00
co-authored by Claude Sonnet 4.6
parent 24bd5cf9dc
commit 32c4def075
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Cargo.toml
+1 -1
View File
@@ -311,7 +311,7 @@ repository = "https://github.com/rustytorch/rustytorch"
[workspace.dependencies] [workspace.dependencies]
# RTX internal crates - Core # RTX internal crates - Core
rtx-backend = { path = "crates/core/rtx-backend", version = "1.0.0" } rtx-backend = { path = "crates/core/rtx-backend", version = "1.0.0" }
# rtx-backend-cuda = { path = "crates/core/rtx-backend-cuda", version = "1.0.0" } # Excluded - requires CUDA/nvcc rtx-backend-cuda = { path = "crates/core/rtx-backend-cuda", version = "1.0.0" }
rtx-backend-metal = { path = "crates/core/rtx-backend-metal", version = "1.0.0" } rtx-backend-metal = { path = "crates/core/rtx-backend-metal", version = "1.0.0" }
rtx-backend-rocm = { path = "crates/core/rtx-backend-rocm", version = "1.0.0" } rtx-backend-rocm = { path = "crates/core/rtx-backend-rocm", version = "1.0.0" }
rtx-backend-sycl = { path = "crates/core/rtx-backend-sycl", version = "0.1.0" } rtx-backend-sycl = { path = "crates/core/rtx-backend-sycl", version = "0.1.0" }
crates/core/rtx-backend-cuda/build.rs
+2 -2
View File
@@ -67,11 +67,11 @@ fn compile_kernel<P: AsRef<Path>>(
} }
// Compile CUDA to PTX // Compile CUDA to PTX
// Use sm_86 (RTX 3050/3090) for compatibility with current GPU // Use sm_120 (RTX 5060 Ti / Blackwell, compute 12.0) for the target GPU
// Use PTX ISA 7.5 for driver compatibility (CUDA 11.x compatible) // Use PTX ISA 7.5 for driver compatibility (CUDA 11.x compatible)
let output = Command::new("nvcc") let output = Command::new("nvcc")
.arg("--ptx") // Generate PTX intermediate representation .arg("--ptx") // Generate PTX intermediate representation
.arg("--gpu-architecture=sm_86") // RTX 3050/3090 (Ampere) .arg("--gpu-architecture=sm_120") // RTX 5060 Ti / Blackwell (compute 12.0)
.arg("--allow-unsupported-compiler") // Allow GCC 13+ (needed for Ubuntu 24.04) .arg("--allow-unsupported-compiler") // Allow GCC 13+ (needed for Ubuntu 24.04)
.arg("-Xptxas") .arg("-Xptxas")
.arg("--allow-expensive-optimizations=true") .arg("--allow-expensive-optimizations=true")