You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Installing CubeCL is straightforward. It’s available as a Rust crate, and you can add it to your project by updating your Cargo.toml:
2
+
3
+
Installing CubeCL is straightforward. It’s available as a Rust crate, and you can add it to your
4
+
project by updating your Cargo.toml:
3
5
4
6
```toml
5
7
[dependencies]
@@ -9,14 +11,17 @@ cubecl = {
9
11
}
10
12
```
11
13
12
-
The more challenging aspect is ensuring that you have the necessary drivers to run the selected runtime.
14
+
The more challenging aspect is ensuring that you have the necessary drivers to run the selected
15
+
runtime.
13
16
14
-
CubeCL supports multiple GPU runtimes, each requiring specific drivers or frameworks. Enable the appropriate feature flag in Cargo.toml and ensure the corresponding drivers are installed.
17
+
CubeCL supports multiple GPU and CPU runtimes, each requiring specific drivers or frameworks. Enable
18
+
the appropriate feature flag in Cargo.toml and ensure the corresponding drivers are installed.
15
19
16
-
| Platform | Runtime | Supported OS | Requirements | Installation/Notes | Feature Flag |
| WebGPU | wgpu | Linux, Windows, macOS, wasm | Vulkan drivers (typically pre-installed on modern OSes) | On linux install the vulkan driver. | wgpu |
19
-
| CUDA | CUDA | Linux, Windows | NVIDIA CUDA drivers and toolkit | Download and install from the NVIDIA CUDA Downloads page. Verify installation with nvidia-smi. | cuda |
20
-
| ROCm | HIP | Linux, Windows | AMD ROCm framework | Linux: Follow the ROCm Linux Quick Start. Windows: See the ROCm Windows Installation Guide. | hip |
21
-
| Metal | wgpu | macOS | Apple device with Metal support (macOS 10.13 or later) | No additional drivers needed; Metal is built into macOS. | wgpu-msl |
22
-
| Vulkan | wgpu | Linux, Windows | Vulkan drivers | On linux install via package manager, on windows it is typically included with GPU drivers (NVIDIA/AMD). | wgpu-spirv |
20
+
| Platform | Device Type | Runtime | Supported OS | Requirements | Installation/Notes | Feature Flag |
| WebGPU | GPU | wgpu | Linux, Windows, macOS, wasm | Vulkan drivers (typically pre-installed on modern OSes) | On linux install the vulkan driver. | wgpu |
23
+
| CUDA | GPU | CUDA | Linux, Windows | NVIDIA CUDA drivers and toolkit | Download and install from the NVIDIA CUDA Downloads page. Verify installation with nvidia-smi. | cuda |
24
+
| ROCm | GPU | HIP | Linux, Windows | AMD ROCm framework | Linux: Follow the ROCm Linux Quick Start. Windows: See the ROCm Windows Installation Guide. | hip |
25
+
| Metal | GPU | wgpu | macOS | Apple device with Metal support (macOS 10.13 or later) | No additional drivers needed; Metal is built into macOS. | wgpu-msl |
26
+
| Vulkan | GPU | wgpu | Linux, Windows | Vulkan drivers | On linux install via package manager, on windows it is typically included with GPU drivers (NVIDIA/AMD). | wgpu-spirv |
27
+
| LLVM | CPU | Rust Std | LLVM supported targets | LLVM bundle | Automatically installed when compiling | cpu |
0 commit comments