Update Readme + Cuda version (#87)

README.md

@@ -4,8 +4,14 @@
 <br />
 <br />
 
-[![Rust Version](https://img.shields.io/badge/Rust-1.75.0+-blue)](https://releases.rs/docs/1.75.0)
+[![Discord](https://img.shields.io/discord/1038839012602941528.svg?color=7289da&&logo=discord)](https://discord.gg/KSBSPhAUCc)
+[![Current Crates.io Version](https://img.shields.io/crates/v/cubecl.svg)](https://crates.io/crates/cubecl)
+[![Test Status](https://github.com/tracel-ai/cubecl/actions/workflows/ci.yml/badge.svg)](https://github.com/tracel-ai/cubecl/actions/workflows/test.yml)
+[![Rust Version](https://img.shields.io/badge/Rust-1.79.0+-blue)](https://releases.rs/docs/1.79.0)
 ![license](https://shields.io/badge/license-MIT%2FApache--2.0-blue)
+<br />
+[![NVIDIA](https://img.shields.io/badge/nvidia-cuda-84b629)](https://github.com/tracel-ai/cubecl/tree/main/crates/cubecl-cuda)
+[![GPU](https://img.shields.io/badge/any_gpu-wgpu-ef604b)](https://github.com/tracel-ai/cubecl/tree/main/crates/cubecl-wgpu)
 
 ---
 

crates/cubecl-cuda/Cargo.toml

@@ -27,7 +27,7 @@ cubecl-core = { path = "../cubecl-core", version = "0.2.0" }
 
 half = { workspace = true }
 bytemuck = { workspace = true }
-cudarc = { version = "=0.11.5", features = ["cuda-12030"] }
+cudarc = { version = "=0.11.5", features = ["cuda-version-from-build-system"] }
 
 log = { workspace = true }
 derive-new = { workspace = true }

crates/cubecl-cuda/README.md

@@ -1,3 +1,11 @@
 # Cuda runtime
 
 The runtime uses the lower level primitives from [cudarc](https://github.com/coreylowman/cudarc) to compile generated CUDA code into a ptx and execute it at runtime.
+
+## Setup
+
+By default, this runtime will attempt to detect the appropriate CUDA version during build time. To specify a fixed CUDA version, you can add `cudarc` to your dependencies with the corresponding feature flag:
+
+```toml
+cudarc = { version = "same as burn", features = ["cuda-11040"] }
+```