diff --git a/.github/workflows/build_kernel_macos.yaml b/.github/workflows/build_kernel_macos.yaml index aaa88787..571e8009 100644 --- a/.github/workflows/build_kernel_macos.yaml +++ b/.github/workflows/build_kernel_macos.yaml @@ -28,7 +28,7 @@ jobs: # For now we only test that there are no regressions in building macOS # kernels. Also run tests once we have a macOS runner. - name: Build relu kernel - run: ( cd examples/kernels/relu && nix build .\#redistributable.torch29-metal-aarch64-darwin -L ) + run: ( cd examples/kernels/relu && nix build .\#redistributable.torch210-metal-aarch64-darwin -L ) - name: Build relu metal cpp kernel - run: ( cd examples/kernels/relu-metal-cpp && nix build .\#redistributable.torch29-metal-aarch64-darwin -L ) + run: ( cd examples/kernels/relu-metal-cpp && nix build .\#redistributable.torch210-metal-aarch64-darwin -L ) diff --git a/docs/source/builder/build-variants.md b/docs/source/builder/build-variants.md index 10f6bd0c..7d33dd62 100644 --- a/docs/source/builder/build-variants.md +++ b/docs/source/builder/build-variants.md @@ -8,42 +8,42 @@ available. This list will be updated as new PyTorch versions are released. ## CPU aarch64-darwin - `torch210-cpu-aarch64-darwin` -- `torch29-cpu-aarch64-darwin` +- `torch211-cpu-aarch64-darwin` ## Metal aarch64-darwin - `torch210-metal-aarch64-darwin` -- `torch29-metal-aarch64-darwin` +- `torch211-metal-aarch64-darwin` ## CPU aarch64-linux - `torch210-cxx11-cpu-aarch64-linux` -- `torch29-cxx11-cpu-aarch64-linux` +- `torch211-cxx11-cpu-aarch64-linux` ## CUDA aarch64-linux - `torch210-cxx11-cu126-aarch64-linux` - `torch210-cxx11-cu128-aarch64-linux` - `torch210-cxx11-cu130-aarch64-linux` -- `torch29-cxx11-cu126-aarch64-linux` -- `torch29-cxx11-cu128-aarch64-linux` +- `torch211-cxx11-cu126-aarch64-linux` +- `torch211-cxx11-cu128-aarch64-linux` +- `torch211-cxx11-cu130-aarch64-linux` - `torch29-cxx11-cu129-aarch64-linux` -- `torch29-cxx11-cu130-aarch64-linux` ## CPU x86_64-linux - `torch210-cxx11-cpu-x86_64-linux` -- `torch29-cxx11-cpu-x86_64-linux` +- `torch211-cxx11-cpu-x86_64-linux` ## CUDA x86_64-linux - `torch210-cxx11-cu126-x86_64-linux` - `torch210-cxx11-cu128-x86_64-linux` - `torch210-cxx11-cu130-x86_64-linux` -- `torch29-cxx11-cu126-x86_64-linux` -- `torch29-cxx11-cu128-x86_64-linux` +- `torch211-cxx11-cu126-x86_64-linux` +- `torch211-cxx11-cu128-x86_64-linux` +- `torch211-cxx11-cu130-x86_64-linux` - `torch29-cxx11-cu129-x86_64-linux` -- `torch29-cxx11-cu130-x86_64-linux` ## ROCm x86_64-linux diff --git a/examples/kernels/flake.nix b/examples/kernels/flake.nix index 919fe6ad..0319a2f7 100644 --- a/examples/kernels/flake.nix +++ b/examples/kernels/flake.nix @@ -15,7 +15,7 @@ inherit (kernel-builder.inputs.nixpkgs) lib; cudaVersion = "cu126"; - torchVersion = "29"; + torchVersion = "210"; tvmFfiVersion = "01"; # All example kernels to build in CI. @@ -79,7 +79,7 @@ drv = sys: out: out.packages.${sys}.default; torchVersions = _defaultVersions: [ { - torchVersion = "2.9"; + torchVersion = "2.10"; cudaVersion = "12.8"; systems = [ "x86_64-linux" diff --git a/examples/kernels/relu-specific-torch/flake.nix b/examples/kernels/relu-specific-torch/flake.nix index edaf0d69..2cc2aa3b 100644 --- a/examples/kernels/relu-specific-torch/flake.nix +++ b/examples/kernels/relu-specific-torch/flake.nix @@ -15,7 +15,7 @@ path = ./.; torchVersions = defaultVersions: [ { - torchVersion = "2.9"; + torchVersion = "2.10"; cudaVersion = "12.8"; systems = [ "x86_64-linux" diff --git a/examples/kernels/relu-torch-bounds/build.toml b/examples/kernels/relu-torch-bounds/build.toml index be8b5396..fcb0ba3d 100644 --- a/examples/kernels/relu-torch-bounds/build.toml +++ b/examples/kernels/relu-torch-bounds/build.toml @@ -6,8 +6,8 @@ backends = [ ] [torch] -minver = "2.9" -maxver = "2.9" +minver = "2.10" +maxver = "2.10" src = [ "torch-ext/torch_binding.cpp", "torch-ext/torch_binding.h", diff --git a/nix-builder/build-variants.json b/nix-builder/build-variants.json index fdfdde08..4359ad76 100644 --- a/nix-builder/build-variants.json +++ b/nix-builder/build-variants.json @@ -2,41 +2,41 @@ "aarch64-darwin": { "cpu": [ "torch210-cpu-aarch64-darwin", - "torch29-cpu-aarch64-darwin" + "torch211-cpu-aarch64-darwin" ], "metal": [ "torch210-metal-aarch64-darwin", - "torch29-metal-aarch64-darwin" + "torch211-metal-aarch64-darwin" ] }, "aarch64-linux": { "cpu": [ "torch210-cxx11-cpu-aarch64-linux", - "torch29-cxx11-cpu-aarch64-linux" + "torch211-cxx11-cpu-aarch64-linux" ], "cuda": [ "torch210-cxx11-cu126-aarch64-linux", "torch210-cxx11-cu128-aarch64-linux", "torch210-cxx11-cu130-aarch64-linux", - "torch29-cxx11-cu126-aarch64-linux", - "torch29-cxx11-cu128-aarch64-linux", - "torch29-cxx11-cu129-aarch64-linux", - "torch29-cxx11-cu130-aarch64-linux" + "torch211-cxx11-cu126-aarch64-linux", + "torch211-cxx11-cu128-aarch64-linux", + "torch211-cxx11-cu130-aarch64-linux", + "torch29-cxx11-cu129-aarch64-linux" ] }, "x86_64-linux": { "cpu": [ "torch210-cxx11-cpu-x86_64-linux", - "torch29-cxx11-cpu-x86_64-linux" + "torch211-cxx11-cpu-x86_64-linux" ], "cuda": [ "torch210-cxx11-cu126-x86_64-linux", "torch210-cxx11-cu128-x86_64-linux", "torch210-cxx11-cu130-x86_64-linux", - "torch29-cxx11-cu126-x86_64-linux", - "torch29-cxx11-cu128-x86_64-linux", - "torch29-cxx11-cu129-x86_64-linux", - "torch29-cxx11-cu130-x86_64-linux" + "torch211-cxx11-cu126-x86_64-linux", + "torch211-cxx11-cu128-x86_64-linux", + "torch211-cxx11-cu130-x86_64-linux", + "torch29-cxx11-cu129-x86_64-linux" ], "rocm": [ "torch210-cxx11-rocm70-x86_64-linux", diff --git a/nix-builder/lib/checks.nix b/nix-builder/lib/checks.nix index 5566e010..71cbce27 100644 --- a/nix-builder/lib/checks.nix +++ b/nix-builder/lib/checks.nix @@ -12,9 +12,10 @@ let path = ../../examples/kernels/relu-torch-bounds; }; in -assert lib.assertMsg (builtins.all (buildSet: buildSet.torch.version == "2.9.0") kernelBuildSets) '' - Torch minver/maxver filtering does not work. -''; +assert lib.assertMsg (builtins.all (buildSet: buildSet.torch.version == "2.10.0") kernelBuildSets) + '' + Torch minver/maxver filtering does not work. + ''; runCommand "builder-nix-checks" { } '' touch $out '' diff --git a/nix-builder/overlay.nix b/nix-builder/overlay.nix index c2f2839a..b82cdbf9 100644 --- a/nix-builder/overlay.nix +++ b/nix-builder/overlay.nix @@ -114,6 +114,11 @@ in xpuPackages = final.xpuPackages_2025_3; }; + torch-bin_2_11 = mkTorch { + version = "2.11"; + xpuPackages = final.xpuPackages_2025_3; + }; + triton-xpu_2_9 = callPackage ./pkgs/python-modules/triton-xpu { torchVersion = "2.9"; xpuPackages = final.xpuPackages_2025_2; diff --git a/nix-builder/pkgs/python-modules/torch/archs.nix b/nix-builder/pkgs/python-modules/torch/archs.nix index 56304771..3f268db4 100644 --- a/nix-builder/pkgs/python-modules/torch/archs.nix +++ b/nix-builder/pkgs/python-modules/torch/archs.nix @@ -176,4 +176,46 @@ ]; }; + "2.11" = { + # https://github.com/pytorch/pytorch/blob/release/2.11/.ci/manywheel/build_cuda.sh + capsPerCudaVersion = { + "13.0" = [ + "7.5" + "8.0" + "8.6" + "9.0" + "10.0" + "11.0" + "12.0" + ]; + # NOTE: 12.9 does not seem to be in RC builds, check if needed for final release. + # https://download.pytorch.org/whl/test/torch/ + "12.9" = [ + "7.0" + "7.5" + "8.0" + "9.0" + "10.0" + "12.0" + ]; + "12.8" = [ + "7.0" + "7.5" + "8.0" + "8.6" + "9.0" + "10.0" + "12.0" + ]; + "12.6" = [ + "5.0" + "6.0" + "7.0" + "7.5" + "8.0" + "8.6" + "9.0" + ]; + }; + }; } diff --git a/nix-builder/pkgs/python-modules/torch/binary/generic.nix b/nix-builder/pkgs/python-modules/torch/binary/generic.nix index 5f46fcc5..fa5b8240 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/generic.nix +++ b/nix-builder/pkgs/python-modules/torch/binary/generic.nix @@ -230,6 +230,7 @@ buildPythonPackage { # the Nix store paths of the framework into the Torch libraries.. pythonRemoveWheelDeps = lib.optionals cudaSupport [ + "cuda-toolkit" "nvidia-cuda-runtime" "nvidia-cuda-nvrtc" "nvidia-cuda-cupti" diff --git a/nix-builder/pkgs/python-modules/torch/binary/torch-versions-hash.json b/nix-builder/pkgs/python-modules/torch/binary/torch-versions-hash.json index 1fd8305f..615ccb14 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/torch-versions-hash.json +++ b/nix-builder/pkgs/python-modules/torch/binary/torch-versions-hash.json @@ -1,26 +1,11 @@ { "2.9": { "x86_64-linux": { - "cu126": { - "url": "https://download.pytorch.org/whl/cu126/torch-2.9.0%2Bcu126-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-DEW2Nu1sAszuYk7FPYRMMrHbm+sj1hnzfEEVJGeQTNU=", - "version": "2.9.0" - }, - "cu128": { - "url": "https://download.pytorch.org/whl/cu128/torch-2.9.0%2Bcu128-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-l97wCH+O8XG5AC6lALr/3UQMe91VnCPDi7+HgbZ+k2Q=", - "version": "2.9.0" - }, "cu129": { "url": "https://download.pytorch.org/whl/cu129/torch-2.9.0%2Bcu129-cp313-cp313-manylinux_2_28_x86_64.whl", "hash": "sha256-a6V77xFJM5fBUddVM0CSKQQSkE5GyF7YYnegO7JP0To=", "version": "2.9.0" }, - "cu130": { - "url": "https://download.pytorch.org/whl/cu130/torch-2.9.0%2Bcu130-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-kqktts3jjQUiDB996Rzqz/AgVGOGxbegomjcquF7XBg=", - "version": "2.9.0" - }, "rocm63": { "url": "https://download.pytorch.org/whl/rocm6.3/torch-2.9.0%2Brocm6.3-cp313-cp313-manylinux_2_28_x86_64.whl", "hash": "sha256-gw3ZjX2JlS7MV8aO6nCyTMAsDO1CSa7z9UzHsW/Wbic=", @@ -31,11 +16,6 @@ "hash": "sha256-yKiHbnxb2P3MOUKJB20rgyla3a1PXKYOo4ZRI9Xcn3Y=", "version": "2.9.0" }, - "cpu": { - "url": "https://download.pytorch.org/whl/cpu/torch-2.9.0%2Bcpu-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-bJshdYRACWPVtNrds3Eex6N3jqshHhhlT7oHbM47hoI=", - "version": "2.9.0" - }, "xpu": { "url": "https://download.pytorch.org/whl/xpu/torch-2.9.0%2Bxpu-cp313-cp313-linux_x86_64.whl", "hash": "sha256-mNagbdf7GFh0NnsYvWCfBfFv3OQUKlmAypRGGUmWXNI=", @@ -43,37 +23,10 @@ } }, "aarch64-linux": { - "cu126": { - "url": "https://download.pytorch.org/whl/cu126/torch-2.9.0%2Bcu126-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-qn+dy/BTDsEKIRRxyK+1Mo1bvKtvFP0beLXm8vFO1hc=", - "version": "2.9.0" - }, - "cu128": { - "url": "https://download.pytorch.org/whl/cu128/torch-2.9.0%2Bcu128-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-TXb3E0WvR/Aix/pV7dDBgQ0Br4ncue3P36/j0qD3prg=", - "version": "2.9.0" - }, "cu129": { "url": "https://download.pytorch.org/whl/cu129/torch-2.9.0%2Bcu129-cp313-cp313-manylinux_2_28_aarch64.whl", "hash": "sha256-wyc11mL4ogcYOM6S72NnjLT4x2Yc3809GwZQO3sAFiY=", "version": "2.9.0" - }, - "cu130": { - "url": "https://download.pytorch.org/whl/cu130/torch-2.9.0%2Bcu130-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-7PPSS9TA5uQlvXeKbembUieeACGmDX6xGrDC1mnz+bA=", - "version": "2.9.0" - }, - "cpu": { - "url": "https://download.pytorch.org/whl/cpu/torch-2.9.0%2Bcpu-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-vkQ42NrX8NWl5U8P7viok0RolOyH8QK7HYLcxFGFQuQ=", - "version": "2.9.0" - } - }, - "aarch64-darwin": { - "cpu": { - "url": "https://download.pytorch.org/whl/cpu/torch-2.9.0-cp313-none-macosx_11_0_arm64.whl", - "hash": "sha256-4kg22Wi1TvTfsFWUABphlYcRrJIkAmKR5OP5L4Om/X8=", - "version": "2.9.0" } } }, @@ -144,5 +97,58 @@ "version": "2.10.0" } } + }, + "2.11": { + "x86_64-linux": { + "cu126": { + "url": "https://download.pytorch.org/whl/cu126/torch-2.11.0%2Bcu126-cp313-cp313-manylinux_2_28_x86_64.whl", + "hash": "sha256-X24qVwmHbdJaX+DU8Kse8zszu2BDH8v+doq2Ro3P6Io=", + "version": "2.11.0" + }, + "cu128": { + "url": "https://download.pytorch.org/whl/cu128/torch-2.11.0%2Bcu128-cp313-cp313-manylinux_2_28_x86_64.whl", + "hash": "sha256-25ZLM8VQNacqs+IWIoevjxzCdgOcZdAVdAzIjCbc7fc=", + "version": "2.11.0" + }, + "cu130": { + "url": "https://download.pytorch.org/whl/cu130/torch-2.11.0%2Bcu130-cp313-cp313-manylinux_2_28_x86_64.whl", + "hash": "sha256-VURht28hIRknx3YFa8sLAPtClyNkeUtobXaOuwtYY2Y=", + "version": "2.11.0" + }, + "cpu": { + "url": "https://download.pytorch.org/whl/cpu/torch-2.11.0%2Bcpu-cp313-cp313-manylinux_2_28_x86_64.whl", + "hash": "sha256-RQJdd1LbxrTHhMA6+u6cXxlzDOCEsuQ/yaL+FnfZ/4Y=", + "version": "2.11.0" + } + }, + "aarch64-linux": { + "cu126": { + "url": "https://download.pytorch.org/whl/cu126/torch-2.11.0%2Bcu126-cp313-cp313-manylinux_2_28_aarch64.whl", + "hash": "sha256-O7v8ftFth4rvm5Go54wDO/Ex/7g1fiP/gJESEGg6MbM=", + "version": "2.11.0" + }, + "cu128": { + "url": "https://download.pytorch.org/whl/cu128/torch-2.11.0%2Bcu128-cp313-cp313-manylinux_2_28_aarch64.whl", + "hash": "sha256-fbNYAQa7oETaW4lQ8/uP5fMZmeqrP2o6oqxdICw2hNI=", + "version": "2.11.0" + }, + "cu130": { + "url": "https://download.pytorch.org/whl/cu130/torch-2.11.0%2Bcu130-cp313-cp313-manylinux_2_28_aarch64.whl", + "hash": "sha256-w9YPeWZrkQHjkUouXewugerINOE8rgvPWelNwaRl91Y=", + "version": "2.11.0" + }, + "cpu": { + "url": "https://download.pytorch.org/whl/cpu/torch-2.11.0%2Bcpu-cp313-cp313-manylinux_2_28_aarch64.whl", + "hash": "sha256-SLPiGjEURazdCyfxODDiHZOt73DUch4FHp8Fm665uPk=", + "version": "2.11.0" + } + }, + "aarch64-darwin": { + "cpu": { + "url": "https://download.pytorch.org/whl/cpu/torch-2.11.0-cp313-cp313-macosx_11_0_arm64.whl", + "hash": "sha256-RC7J3HhZJWT9rWnPC+qp2i+Cq4EMy08TkDhpqQvz8V0=", + "version": "2.11.0" + } + } } } \ No newline at end of file diff --git a/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json b/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json index a0deddbd..ee303a99 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json +++ b/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json @@ -1,24 +1,9 @@ [ - { - "torchVersion": "2.9.0", - "cudaVersion": "12.6", - "systems": ["x86_64-linux", "aarch64-linux"] - }, - { - "torchVersion": "2.9.0", - "cudaVersion": "12.8", - "systems": ["x86_64-linux", "aarch64-linux"] - }, { "torchVersion": "2.9.0", "cudaVersion": "12.9", "systems": ["x86_64-linux", "aarch64-linux"] }, - { - "torchVersion": "2.9.0", - "cudaVersion": "13.0", - "systems": ["x86_64-linux", "aarch64-linux"] - }, { "torchVersion": "2.9.0", "rocmVersion": "6.3", @@ -29,16 +14,6 @@ "rocmVersion": "6.4", "systems": ["x86_64-linux"] }, - { - "torchVersion": "2.9.0", - "metal": true, - "systems": ["aarch64-darwin"] - }, - { - "torchVersion": "2.9.0", - "cpu": true, - "systems": ["aarch64-linux", "x86_64-linux"] - }, { "torchVersion": "2.9.0", "xpuVersion": "2025.2.1", @@ -84,5 +59,31 @@ "torchVersion": "2.10.0", "xpuVersion": "2025.3.1", "systems": ["x86_64-linux"] + }, + + { + "torchVersion": "2.11.0", + "cudaVersion": "12.6", + "systems": ["x86_64-linux", "aarch64-linux"] + }, + { + "torchVersion": "2.11.0", + "cudaVersion": "12.8", + "systems": ["x86_64-linux", "aarch64-linux"] + }, + { + "torchVersion": "2.11.0", + "cudaVersion": "13.0", + "systems": ["x86_64-linux", "aarch64-linux"] + }, + { + "torchVersion": "2.11.0", + "cpu": true, + "systems": ["aarch64-linux", "x86_64-linux"] + }, + { + "torchVersion": "2.11.0", + "metal": true, + "systems": ["aarch64-darwin"] } ] diff --git a/nix-builder/pkgs/python-modules/torch/binary/torch_versions.py b/nix-builder/pkgs/python-modules/torch/binary/torch_versions.py index c3aa154b..7b2cf99d 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/torch_versions.py +++ b/nix-builder/pkgs/python-modules/torch/binary/torch_versions.py @@ -53,7 +53,7 @@ def generate_pytorch_url( if "darwin" in system: framework_dir = "cpu" version_part = torch_version - abi_tag = "none" if "darwin" in system else python_version + abi_tag = "none" if "2.10" in version_part else python_version wheel_name = f"torch-{version_part}-{python_version}-{abi_tag}-{platform}.whl" elif framework_type == "cpu": framework_dir = "cpu" diff --git a/nix-builder/tests/Dockerfile.test-kernel b/nix-builder/tests/Dockerfile.test-kernel index df98e0d9..92b8c485 100644 --- a/nix-builder/tests/Dockerfile.test-kernel +++ b/nix-builder/tests/Dockerfile.test-kernel @@ -3,7 +3,7 @@ ARG PYTHON_VERSION=3.10 # Test with the oldest CUDA version supported by the latest PyTorch. ARG CUDA_VERSION=12.6.0 ARG UBI_VERSION=8 -ARG TORCH_VERSION=2.9.0 +ARG TORCH_VERSION=2.10.0 FROM nvidia/cuda:${CUDA_VERSION}-devel-ubi${UBI_VERSION} as base diff --git a/nix-builder/versions.nix b/nix-builder/versions.nix index 05093f9e..dde47114 100644 --- a/nix-builder/versions.nix +++ b/nix-builder/versions.nix @@ -8,7 +8,7 @@ } { torchVersion = "2.9"; - cudaVersion = "12.6"; + cudaVersion = "12.9"; systems = [ "x86_64-linux" "aarch64-linux" @@ -17,7 +17,20 @@ } { torchVersion = "2.9"; - cudaVersion = "12.8"; + rocmVersion = "6.3.4"; + systems = [ "x86_64-linux" ]; + bundleBuild = true; + } + { + torchVersion = "2.9"; + rocmVersion = "6.4.2"; + systems = [ "x86_64-linux" ]; + bundleBuild = true; + } + + { + torchVersion = "2.10"; + cudaVersion = "12.6"; systems = [ "x86_64-linux" "aarch64-linux" @@ -25,8 +38,8 @@ bundleBuild = true; } { - torchVersion = "2.9"; - cudaVersion = "12.9"; + torchVersion = "2.10"; + cudaVersion = "12.8"; systems = [ "x86_64-linux" "aarch64-linux" @@ -34,7 +47,7 @@ bundleBuild = true; } { - torchVersion = "2.9"; + torchVersion = "2.10"; cudaVersion = "13.0"; systems = [ "x86_64-linux" @@ -43,25 +56,19 @@ bundleBuild = true; } { - torchVersion = "2.9"; - rocmVersion = "6.3.4"; + torchVersion = "2.10"; + rocmVersion = "7.0"; systems = [ "x86_64-linux" ]; bundleBuild = true; } { - torchVersion = "2.9"; - rocmVersion = "6.4.2"; + torchVersion = "2.10"; + rocmVersion = "7.1"; systems = [ "x86_64-linux" ]; bundleBuild = true; } { - torchVersion = "2.9"; - metal = true; - systems = [ "aarch64-darwin" ]; - bundleBuild = true; - } - { - torchVersion = "2.9"; + torchVersion = "2.10"; cpu = true; systems = [ "aarch64-darwin" @@ -70,11 +77,25 @@ ]; bundleBuild = true; } - { torchVersion = "2.10"; - cudaVersion = "12.6"; + metal = true; + systems = [ "aarch64-darwin" ]; + bundleBuild = true; + } + { + torchVersion = "2.10"; + xpuVersion = "2025.3.1"; + systems = [ "x86_64-linux" ]; + bundleBuild = true; + tvmFfi = true; + } + + { + torchVersion = "2.11"; + cpu = true; systems = [ + "aarch64-darwin" "x86_64-linux" "aarch64-linux" ]; @@ -82,8 +103,8 @@ tvmFfi = true; } { - torchVersion = "2.10"; - cudaVersion = "12.8"; + torchVersion = "2.11"; + cudaVersion = "12.6"; systems = [ "x86_64-linux" "aarch64-linux" @@ -92,8 +113,8 @@ tvmFfi = true; } { - torchVersion = "2.10"; - cudaVersion = "13.0"; + torchVersion = "2.11"; + cudaVersion = "12.8"; systems = [ "x86_64-linux" "aarch64-linux" @@ -102,22 +123,9 @@ tvmFfi = true; } { - torchVersion = "2.10"; - rocmVersion = "7.0"; - systems = [ "x86_64-linux" ]; - bundleBuild = true; - } - { - torchVersion = "2.10"; - rocmVersion = "7.1"; - systems = [ "x86_64-linux" ]; - bundleBuild = true; - } - { - torchVersion = "2.10"; - cpu = true; + torchVersion = "2.11"; + cudaVersion = "13.0"; systems = [ - "aarch64-darwin" "x86_64-linux" "aarch64-linux" ]; @@ -125,16 +133,9 @@ tvmFfi = true; } { - torchVersion = "2.10"; + torchVersion = "2.11"; metal = true; systems = [ "aarch64-darwin" ]; bundleBuild = true; } - { - torchVersion = "2.10"; - xpuVersion = "2025.3.1"; - systems = [ "x86_64-linux" ]; - bundleBuild = true; - tvmFfi = true; - } ]