Tegra, deactivation, CUDA, autograd

.ci_support/linux_aarch64_c_stdlib_version2.17channel_targetsconda-forge_maincuda_compiler_version12.9github_actions_labelscirun-openstack-gpu-2xlargeis_rcFalse.yaml → .ci_support/linux_aarch64_c_compiler_version13c_stdlib_version2.17channel_targetsconda-forge_maincuda_compiler_version12.9cxx_compiler_version13github_actions_labelscirun-openstack-gpu-2xlargeis_rcFalse.yaml

@@ -1,3 +1,5 @@
+arm_variant_type:
+- sbsa
 blas_impl:
 - generic
 c_compiler:
@@ -75,6 +77,7 @@ zip_keys:
   - cxx_compiler_version
   - c_stdlib_version
   - cuda_compiler_version
+  - arm_variant_type
 - - channel_targets
   - is_rc
 zlib:

.ci_support/linux_aarch64_c_stdlib_version2.17channel_targetsconda-forge_maincuda_compiler_versionNonegithub_actions_labelscirun-openstack-cpu-xlargeis_rcFalse.yaml → .ci_support/linux_aarch64_c_compiler_version13c_stdlib_version2.17channel_targetsconda-forge_maincuda_compiler_versionNonecxx_compiler_version13github_actions_labelscirun-openstack-cpu-xlargeis_rcFalse.yaml

@@ -1,3 +1,5 @@
+arm_variant_type:
+- sbsa
 blas_impl:
 - generic
 c_compiler:
@@ -75,6 +77,7 @@ zip_keys:
   - cxx_compiler_version
   - c_stdlib_version
   - cuda_compiler_version
+  - arm_variant_type
 - - channel_targets
   - is_rc
 zlib:

.ci_support/linux_aarch64_c_stdlib_version2.28channel_targetsconda-forge_maincuda_compiler_version13.0github_actions_labelscirun-openstack-gpu-2xlargeis_rcFalse.yaml → .ci_support/linux_aarch64_c_compiler_version13c_stdlib_version2.28channel_targetsconda-forge_maincuda_compiler_version13.0cxx_compiler_version13github_actions_labelscirun-openstack-gpu-2xlargeis_rcFalse.yaml

@@ -1,3 +1,5 @@
+arm_variant_type:
+- sbsa
 blas_impl:
 - generic
 c_compiler:
@@ -75,6 +77,7 @@ zip_keys:
   - cxx_compiler_version
   - c_stdlib_version
   - cuda_compiler_version
+  - arm_variant_type
 - - channel_targets
   - is_rc
 zlib:

.ci_support/linux_aarch64_c_compiler_version14c_stdlib_version2.34channel_targetsconda-forge_maincuda_compiler_version12.9cxx_compiler_version14github_actions_labelscirun-openstack-gpu-2xlargeis_rcFalse.yaml

@@ -0,0 +1,84 @@
+arm_variant_type:
+- tegra
+blas_impl:
+- generic
+c_compiler:
+- gcc
+c_compiler_version:
+- '14'
+c_stdlib:
+- sysroot
+c_stdlib_version:
+- '2.34'
+channel_sources:
+- conda-forge
+channel_targets:
+- conda-forge main
+cuda_compiler:
+- cuda-nvcc
+cuda_compiler_version:
+- '12.9'
+cxx_compiler:
+- gxx
+cxx_compiler_version:
+- '14'
+docker_image:
+- quay.io/condaforge/linux-anvil-x86_64:alma9
+fmt:
+- '12.1'
+github_actions_labels:
+- cirun-openstack-gpu-2xlarge
+is_rc:
+- 'False'
+libabseil:
+- '20260107'
+libblas:
+- 3.9.* *netlib
+libcblas:
+- 3.9.* *netlib
+libcudnn_dev:
+- '9'
+liblapack:
+- 3.9.* *netlib
+libmagma_devel:
+- '2.9'
+libmagma_sparse:
+- '2.9'
+libprotobuf:
+- 6.33.5
+libtorch:
+- '2.9'
+mkl:
+- '2025'
+nccl:
+- '2'
+numpy:
+- '2'
+orc:
+- 2.2.2
+pin_run_as_build:
+  python:
+    min_pin: x.x
+    max_pin: x.x
+pybind11_abi:
+- '11'
+python:
+- 3.10.* *_cpython
+- 3.11.* *_cpython
+- 3.12.* *_cpython
+- 3.13.* *_cp313
+- 3.14.* *_cp314
+pytorch:
+- '2.9'
+target_platform:
+- linux-aarch64
+zip_keys:
+- - c_compiler_version
+  - cxx_compiler_version
+  - c_stdlib_version
+  - cuda_compiler_version
+  - arm_variant_type
+- - channel_targets
+  - is_rc
+zlib:
+- '1'

.ci_support/migrations/arm_variant_type.yaml

@@ -0,0 +1,77 @@
+# time stamped to be the day before the CUDA 11.8 migrator
+migrator_ts: 2145852000  # 2037-12-31
+__migrator:
+  # this migration should not be unpaused!
+  # It's intended to be copied on an as-needed basis to feedstocks that want to support tegra
+  paused: true
+  operation: key_add
+  migration_number:
+    1
+  build_number:
+    1
+  override_cbc_keys:
+    - cuda_compiler_stub
+  check_solvable: false
+  primary_key: cuda_compiler_version
+  ordering:
+    arm_variant_type:
+      - None
+      - sbsa
+      - tegra
+  additional_zip_keys:  # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+    - arm_variant_type  # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+  wait_for_migrators:
+    - cuda129
+    - aarch64 and ppc64le addition
+  commit_message: |2
+    Build for NVIDIA Tegra devices and CUDA 12.9
+
+    This migration adds `arm_variant_type=tegra` to the build matrix to support NVIDIA Tegra
+    devices compatible with CUDA 12.9. This migrator is only applicable to the `linux-aarch4`
+    platform because Tegra is specific to that architecture. Non-Tegra ARM
+    devices are assumed to be SBSA-compliant (Server Base System Architecture). The
+    default value of `arm_variant_type` is `sbsa` or it is undefined for non-ARM platforms.
+    Tegra devices compatible with CUDA 13.0 are SBSA-compliant, and do not need a separate
+    build. Only Orin (sm_87) and later devices are supported because earlier Tegra devices are
+    not supported by CUDA 12.9.
+
+    In addition to this migrator, the `arm-variant` package must be added to the build
+    requirements of the recipe in order to constrain the CUDA compiler to the correct variant.
+
+    ```yaml
+    # A fake selector may be needed for conda-build to pick up arm_variant_type as a variant
+    # [arm_variant_type]
+
+    requirements:
+      build:
+        - {{ compiler('cuda') }}
+        - arm-variant * {{ arm_variant_type }}  # [linux and aarch64 and cuda_compiler_version != "None"]
+    ```
+
+    For v1 recipes, the work-around looks as follows:
+    ```yaml
+    context:
+      # ensure arm_variant_type gets detected as a used variable
+      touch_arm_variant_type: ${{ arm_variant_type }}
+    ```
+
+    Please read the conda-forge CUDA recipe guide for more information:
+    https://github.com/conda-forge/cuda-feedstock/blob/main/recipe/doc/recipe_guide.md#building-for-arm-tegra-devices
+
+c_compiler_version:            # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+  - 14                         # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+
+cxx_compiler_version:          # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+  - 14                         # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+
+fortran_compiler_version:      # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+  - 14                         # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+
+cuda_compiler_version:         # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+  - 12.9                       # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+
+c_stdlib_version:              # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+  - 2.34                       # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+
+arm_variant_type:              # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+  - tegra                      # [linux and aarch64 and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]

.ci_support/win_64_channel_targetsconda-forge_maincuda_compiler_version12.8is_rcFalse.yaml

@@ -32,8 +32,6 @@ libprotobuf:
 - 6.33.5
 libtorch:
 - '2.9'
-magma:
-- '2.9'
 mkl:
 - '2025'
 mkl_devel:

.ci_support/win_64_channel_targetsconda-forge_maincuda_compiler_version13.0is_rcFalse.yaml

@@ -32,8 +32,6 @@ libprotobuf:
 - 6.33.5
 libtorch:
 - '2.9'
-magma:
-- '2.9'
 mkl:
 - '2025'
 mkl_devel:

.ci_support/win_64_channel_targetsconda-forge_maincuda_compiler_versionNoneis_rcFalse.yaml

@@ -32,8 +32,6 @@ libprotobuf:
 - 6.33.5
 libtorch:
 - '2.9'
-magma:
-- '2.9'
 mkl:
 - '2025'
 mkl_devel:

.github/workflows/conda-build.yml

@@ -61,24 +61,31 @@ jobs:
             runs_on: ['cirun-openstack-gpu-2xlarge--${{ github.run_id }}-linux_64_blas_implmklc_stdlib_version2.2_ha4aa5d9c', 'linux', 'x64', 'self-hosted']
             DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
             CONDA_FORGE_DOCKER_RUN_ARGS: "--gpus all"
-          - CONFIG: linux_aarch64_c_stdlib_version2.17channel_targetsconda-forge_maincuda_compiler_version12.9github_actions_labelscirun-openstack-gpu-2xlargeis_rcFalse
-            SHORT_CONFIG: linux_aarch64_c_stdlib_version2.17channe_h7b39afa9
+          - CONFIG: linux_aarch64_c_compiler_version13c_stdlib_version2.17channel_targetsconda-forge_maincuda_compiler_version12.9cxx_compiler_version13github_actions_labelscirun-openstack-gpu-2xlargeis_rcFalse
+            SHORT_CONFIG: linux_aarch64_c_compiler_version13c_stdl_hae63c33e
             UPLOAD_PACKAGES: True
             os: ubuntu
-            runs_on: ['cirun-openstack-gpu-2xlarge--${{ github.run_id }}-linux_aarch64_c_stdlib_version2.17channe_h7b39afa9', 'linux', 'x64', 'self-hosted']
+            runs_on: ['cirun-openstack-gpu-2xlarge--${{ github.run_id }}-linux_aarch64_c_compiler_version13c_stdl_hae63c33e', 'linux', 'x64', 'self-hosted']
             DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
             CONDA_FORGE_DOCKER_RUN_ARGS: "--gpus all"
-          - CONFIG: linux_aarch64_c_stdlib_version2.17channel_targetsconda-forge_maincuda_compiler_versionNonegithub_actions_labelscirun-openstack-cpu-xlargeis_rcFalse
-            SHORT_CONFIG: linux_aarch64_c_stdlib_version2.17channe_hbde90b87
+          - CONFIG: linux_aarch64_c_compiler_version13c_stdlib_version2.17channel_targetsconda-forge_maincuda_compiler_versionNonecxx_compiler_version13github_actions_labelscirun-openstack-cpu-xlargeis_rcFalse
+            SHORT_CONFIG: linux_aarch64_c_compiler_version13c_stdl_h33a70202
             UPLOAD_PACKAGES: True
             os: ubuntu
-            runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_aarch64_c_stdlib_version2.17channe_hbde90b87', 'linux', 'x64', 'self-hosted']
+            runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_aarch64_c_compiler_version13c_stdl_h33a70202', 'linux', 'x64', 'self-hosted']
             DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
-          - CONFIG: linux_aarch64_c_stdlib_version2.28channel_targetsconda-forge_maincuda_compiler_version13.0github_actions_labelscirun-openstack-gpu-2xlargeis_rcFalse
-            SHORT_CONFIG: linux_aarch64_c_stdlib_version2.28channe_h4c53fe26
+          - CONFIG: linux_aarch64_c_compiler_version13c_stdlib_version2.28channel_targetsconda-forge_maincuda_compiler_version13.0cxx_compiler_version13github_actions_labelscirun-openstack-gpu-2xlargeis_rcFalse
+            SHORT_CONFIG: linux_aarch64_c_compiler_version13c_stdl_h9855fad4
             UPLOAD_PACKAGES: True
             os: ubuntu
-            runs_on: ['cirun-openstack-gpu-2xlarge--${{ github.run_id }}-linux_aarch64_c_stdlib_version2.28channe_h4c53fe26', 'linux', 'x64', 'self-hosted']
+            runs_on: ['cirun-openstack-gpu-2xlarge--${{ github.run_id }}-linux_aarch64_c_compiler_version13c_stdl_h9855fad4', 'linux', 'x64', 'self-hosted']
+            DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
+            CONDA_FORGE_DOCKER_RUN_ARGS: "--gpus all"
+          - CONFIG: linux_aarch64_c_compiler_version14c_stdlib_version2.34channel_targetsconda-forge_maincuda_compiler_version12.9cxx_compiler_version14github_actions_labelscirun-openstack-gpu-2xlargeis_rcFalse
+            SHORT_CONFIG: linux_aarch64_c_compiler_version14c_stdl_hbec9a05b
+            UPLOAD_PACKAGES: True
+            os: ubuntu
+            runs_on: ['cirun-openstack-gpu-2xlarge--${{ github.run_id }}-linux_aarch64_c_compiler_version14c_stdl_hbec9a05b', 'linux', 'x64', 'self-hosted']
             DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
             CONDA_FORGE_DOCKER_RUN_ARGS: "--gpus all"
           - CONFIG: osx_64_blas_implgenericchannel_targetsconda-forge_mainis_rcFalse
@@ -114,7 +121,7 @@ jobs:
     steps:
 
     - name: Checkout code
-      uses: actions/checkout@0ad4b8fadaa221de15dcec353f45205ec38ea70b # v4.1.4
+      uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
 
     - name: Build on Linux
       id: build-linux
@@ -281,7 +288,7 @@ jobs:
         fi
 
     - name: Store conda build artifacts
-      uses: actions/upload-artifact@ea165f8d65b6e75b540449e92b4886f43607fa02 # v4.6.2
+      uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f # v6.0.0
       if: ${{ always() && steps.prepare-artifacts.outcome == 'success' }}
       with:
         name: ${{ steps.prepare-artifacts.outputs.BLD_ARTIFACT_NAME }}
@@ -290,7 +297,7 @@ jobs:
       continue-on-error: true
 
     - name: Store conda build environment artifacts
-      uses: actions/upload-artifact@ea165f8d65b6e75b540449e92b4886f43607fa02 # v4.6.2
+      uses: actions/upload-artifact@b7c566a772e6b6bfb58ed0dc250532a479d7789f # v6.0.0
       # only relevant if build failed, see above
       if: ${{ always() && steps.determine-status.outputs.status == 'failure' && steps.prepare-artifacts.outcome == 'success' }}
       with:

conda-forge.yml

@@ -10,6 +10,9 @@ build_platform:
   osx_64: osx_arm64
 conda_build:
   pkg_format: '2'
+  # use highest possible compression (smaller package -> less upload time) to try to
+  # combat timouts on windows see https://github.com/conda/infrastructure/issues/1159
+  zstd_compression_level: 22
 conda_forge_output_validation: true
 github:
   branch_name: main