Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .github/workflows/test_kernels.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -29,11 +29,11 @@ jobs:
max-parallel: 4
matrix:
python-version: ["3.10", "3.14"]
torch-version: ["2.11.0", "2.12.0"]
torch-version: ["2.12.0", "2.13.0"]

env:
UV_PYTHON_PREFERENCE: only-managed
CHECK_COVERAGE: ${{ matrix.python-version == '3.10' && matrix.torch-version == '2.12.0' }}
CHECK_COVERAGE: ${{ matrix.python-version == '3.10' && matrix.torch-version == '2.13.0' }}

steps:
- name: Checkout code
Expand Down
13 changes: 0 additions & 13 deletions docs/source/builder/build-variants.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,27 +7,21 @@ available. This list will be updated as new PyTorch versions are released.

## CPU aarch64-darwin

- `torch211-cpu-aarch64-darwin`
- `torch212-cpu-aarch64-darwin`
- `torch213-cpu-aarch64-darwin`

## Metal aarch64-darwin

- `torch211-metal-aarch64-darwin`
- `torch212-metal-aarch64-darwin`
- `torch213-metal-aarch64-darwin`

## CPU aarch64-linux

- `torch211-cxx11-cpu-aarch64-linux`
- `torch212-cxx11-cpu-aarch64-linux`
- `torch213-cxx11-cpu-aarch64-linux`

## CUDA aarch64-linux

- `torch211-cxx11-cu126-aarch64-linux`
- `torch211-cxx11-cu128-aarch64-linux`
- `torch211-cxx11-cu130-aarch64-linux`
- `torch212-cxx11-cu126-aarch64-linux`
- `torch212-cxx11-cu130-aarch64-linux`
- `torch212-cxx11-cu132-aarch64-linux`
Expand All @@ -37,15 +31,11 @@ available. This list will be updated as new PyTorch versions are released.

## CPU x86_64-linux

- `torch211-cxx11-cpu-x86_64-linux`
- `torch212-cxx11-cpu-x86_64-linux`
- `torch213-cxx11-cpu-x86_64-linux`

## CUDA x86_64-linux

- `torch211-cxx11-cu126-x86_64-linux`
- `torch211-cxx11-cu128-x86_64-linux`
- `torch211-cxx11-cu130-x86_64-linux`
- `torch212-cxx11-cu126-x86_64-linux`
- `torch212-cxx11-cu130-x86_64-linux`
- `torch212-cxx11-cu132-x86_64-linux`
Expand All @@ -55,16 +45,13 @@ available. This list will be updated as new PyTorch versions are released.

## ROCm x86_64-linux

- `torch211-cxx11-rocm71-x86_64-linux`
- `torch211-cxx11-rocm72-x86_64-linux`
- `torch212-cxx11-rocm71-x86_64-linux`
- `torch212-cxx11-rocm72-x86_64-linux`
- `torch213-cxx11-rocm71-x86_64-linux`
- `torch213-cxx11-rocm72-x86_64-linux`

## XPU x86_64-linux

- `torch211-cxx11-xpu20253-x86_64-linux`
- `torch212-cxx11-xpu20253-x86_64-linux`
- `torch213-cxx11-xpu20260-x86_64-linux`

Expand Down
2 changes: 1 addition & 1 deletion docs/source/builder/build.md
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,7 @@ using:

```bash
$ rm -rf .venv # Remove existing venv if any.
$ kernel-builder devshell --variant torch211-cxx11-rocm71-x86_64-linux
$ kernel-builder devshell --variant torch212-cxx11-rocm71-x86_64-linux
```

For an editor-driven workflow with `direnv` activating the devshell on
Expand Down
10 changes: 5 additions & 5 deletions docs/source/builder/ide-setup.md
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ the `Creating new venv environment in path: './.venv'` line from the
To pin a non-default build variant, name it explicitly:

```bash
$ echo 'use flake .#devShells.torch211-cxx11-rocm71-x86_64-linux' > .envrc
$ echo 'use flake .#devShells.torch212-cxx11-rocm71-x86_64-linux' > .envrc
$ direnv allow
```

Expand Down Expand Up @@ -145,7 +145,7 @@ Add `.vscode/settings.json` (do not commit):
"clangd.arguments": ["--compile-commands-dir=${workspaceFolder}"],

// Microsoft C/C++ extension
"C_Cpp.default.compileCommands": "${workspaceFolder}/compile_commands.json"
"C_Cpp.default.compileCommands": "${workspaceFolder}/compile_commands.json",
}
```

Expand Down Expand Up @@ -182,13 +182,13 @@ variant. For example:

```bash
# CUDA 13.0
use flake .#devShells.torch211-cxx11-cu130-x86_64-linux
use flake .#devShells.torch212-cxx11-cu130-x86_64-linux

# ROCm 7.1
use flake .#devShells.torch211-cxx11-rocm71-x86_64-linux
use flake .#devShells.torch212-cxx11-rocm71-x86_64-linux

# XPU
use flake .#devShells.torch211-cxx11-xpu20253-x86_64-linux
use flake .#devShells.torch212-cxx11-xpu20253-x86_64-linux
```

Remove `.venv/` first if it was created against a different variant,
Expand Down
10 changes: 4 additions & 6 deletions examples/kernels/flake.nix
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
cudaVersion = "cu126";
rocmVersion = "rocm71";
xpuVersion = "xpu20253";
torchVersion = "211";
torchVersion = "212";
tvmFfiVersion = "01";

# All example kernels to build in CI.
Expand Down Expand Up @@ -70,9 +70,7 @@
{
name = "relu-torch-stable-abi-kernel";
path = ./relu-torch-stable-abi;
drv =
sys: out:
out.packages.${sys}.redistributable.${"torch-stable-abi${torchVersion}-${cudaVersion}-${sys}"};
drv = sys: out: out.packages.${sys}.redistributable.${"torch-stable-abi211-${cudaVersion}-${sys}"};
}
{
name = "relu-tvm-ffi-kernel";
Expand Down Expand Up @@ -127,8 +125,8 @@
drv = sys: out: out.packages.${sys}.default;
torchVersions = _defaultVersions: [
{
torchVersion = "2.11";
cudaVersion = "12.8";
torchVersion = "2.12";
cudaVersion = "12.6";
systems = [
"x86_64-linux"
"aarch64-linux"
Expand Down
4 changes: 2 additions & 2 deletions examples/kernels/relu-specific-torch/flake.nix
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,8 @@
path = ./.;
torchVersions = defaultVersions: [
{
torchVersion = "2.11";
cudaVersion = "12.8";
torchVersion = "2.12";
cudaVersion = "12.6";
systems = [
"x86_64-linux"
"aarch64-linux"
Expand Down
Loading
Loading