diff --git a/.azure-pipelines/azure-pipelines-linux.yml b/.azure-pipelines/azure-pipelines-linux.yml index 7932850..0eb853a 100755 --- a/.azure-pipelines/azure-pipelines-linux.yml +++ b/.azure-pipelines/azure-pipelines-linux.yml @@ -8,20 +8,36 @@ jobs: vmImage: ubuntu-latest strategy: matrix: - linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.10.____cpython: - CONFIG: linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.10.____cpython + linux_64_cuda_compiler_version12.9python3.10.____cpython: + CONFIG: linux_64_cuda_compiler_version12.9python3.10.____cpython UPLOAD_PACKAGES: 'True' DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9 - linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.11.____cpython: - CONFIG: linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.11.____cpython + linux_64_cuda_compiler_version12.9python3.11.____cpython: + CONFIG: linux_64_cuda_compiler_version12.9python3.11.____cpython UPLOAD_PACKAGES: 'True' DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9 - linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.12.____cpython: - CONFIG: linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.12.____cpython + linux_64_cuda_compiler_version12.9python3.12.____cpython: + CONFIG: linux_64_cuda_compiler_version12.9python3.12.____cpython UPLOAD_PACKAGES: 'True' DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9 - linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.13.____cp313: - CONFIG: linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.13.____cp313 + linux_64_cuda_compiler_version12.9python3.13.____cp313: + CONFIG: linux_64_cuda_compiler_version12.9python3.13.____cp313 + UPLOAD_PACKAGES: 'True' + DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9 + linux_aarch64_cuda_compiler_version12.9python3.10.____cpython: + CONFIG: linux_aarch64_cuda_compiler_version12.9python3.10.____cpython + UPLOAD_PACKAGES: 'True' + DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9 + linux_aarch64_cuda_compiler_version12.9python3.11.____cpython: + CONFIG: linux_aarch64_cuda_compiler_version12.9python3.11.____cpython + UPLOAD_PACKAGES: 'True' + DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9 + linux_aarch64_cuda_compiler_version12.9python3.12.____cpython: + CONFIG: linux_aarch64_cuda_compiler_version12.9python3.12.____cpython + UPLOAD_PACKAGES: 'True' + DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9 + linux_aarch64_cuda_compiler_version12.9python3.13.____cp313: + CONFIG: linux_aarch64_cuda_compiler_version12.9python3.13.____cp313 UPLOAD_PACKAGES: 'True' DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9 timeoutInMinutes: 360 diff --git a/.ci_support/linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.10.____cpython.yaml b/.ci_support/linux_64_cuda_compiler_version12.9python3.10.____cpython.yaml similarity index 95% rename from .ci_support/linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.10.____cpython.yaml rename to .ci_support/linux_64_cuda_compiler_version12.9python3.10.____cpython.yaml index 4843e52..012f107 100644 --- a/.ci_support/linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.10.____cpython.yaml +++ b/.ci_support/linux_64_cuda_compiler_version12.9python3.10.____cpython.yaml @@ -1,7 +1,7 @@ c_compiler: - gcc c_compiler_version: -- '13' +- '14' c_stdlib: - sysroot c_stdlib_version: @@ -15,11 +15,11 @@ channel_targets: cuda_compiler: - cuda-nvcc cuda_compiler_version: -- '12.6' +- '12.9' cxx_compiler: - gxx cxx_compiler_version: -- '13' +- '14' docker_image: - quay.io/condaforge/linux-anvil-x86_64:alma9 pin_run_as_build: diff --git a/.ci_support/linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.11.____cpython.yaml b/.ci_support/linux_64_cuda_compiler_version12.9python3.11.____cpython.yaml similarity index 95% rename from .ci_support/linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.11.____cpython.yaml rename to .ci_support/linux_64_cuda_compiler_version12.9python3.11.____cpython.yaml index 145cbe3..1b7bbbd 100644 --- a/.ci_support/linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.11.____cpython.yaml +++ b/.ci_support/linux_64_cuda_compiler_version12.9python3.11.____cpython.yaml @@ -1,7 +1,7 @@ c_compiler: - gcc c_compiler_version: -- '13' +- '14' c_stdlib: - sysroot c_stdlib_version: @@ -15,11 +15,11 @@ channel_targets: cuda_compiler: - cuda-nvcc cuda_compiler_version: -- '12.6' +- '12.9' cxx_compiler: - gxx cxx_compiler_version: -- '13' +- '14' docker_image: - quay.io/condaforge/linux-anvil-x86_64:alma9 pin_run_as_build: diff --git a/.ci_support/linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.12.____cpython.yaml b/.ci_support/linux_64_cuda_compiler_version12.9python3.12.____cpython.yaml similarity index 95% rename from .ci_support/linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.12.____cpython.yaml rename to .ci_support/linux_64_cuda_compiler_version12.9python3.12.____cpython.yaml index 5378c4e..7783abc 100644 --- a/.ci_support/linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.12.____cpython.yaml +++ b/.ci_support/linux_64_cuda_compiler_version12.9python3.12.____cpython.yaml @@ -1,7 +1,7 @@ c_compiler: - gcc c_compiler_version: -- '13' +- '14' c_stdlib: - sysroot c_stdlib_version: @@ -15,11 +15,11 @@ channel_targets: cuda_compiler: - cuda-nvcc cuda_compiler_version: -- '12.6' +- '12.9' cxx_compiler: - gxx cxx_compiler_version: -- '13' +- '14' docker_image: - quay.io/condaforge/linux-anvil-x86_64:alma9 pin_run_as_build: diff --git a/.ci_support/linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.13.____cp313.yaml b/.ci_support/linux_64_cuda_compiler_version12.9python3.13.____cp313.yaml similarity index 95% rename from .ci_support/linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.13.____cp313.yaml rename to .ci_support/linux_64_cuda_compiler_version12.9python3.13.____cp313.yaml index 76b75d5..857c3d9 100644 --- a/.ci_support/linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.13.____cp313.yaml +++ b/.ci_support/linux_64_cuda_compiler_version12.9python3.13.____cp313.yaml @@ -1,7 +1,7 @@ c_compiler: - gcc c_compiler_version: -- '13' +- '14' c_stdlib: - sysroot c_stdlib_version: @@ -15,11 +15,11 @@ channel_targets: cuda_compiler: - cuda-nvcc cuda_compiler_version: -- '12.6' +- '12.9' cxx_compiler: - gxx cxx_compiler_version: -- '13' +- '14' docker_image: - quay.io/condaforge/linux-anvil-x86_64:alma9 pin_run_as_build: diff --git a/.ci_support/linux_aarch64_cuda_compiler_version12.9python3.10.____cpython.yaml b/.ci_support/linux_aarch64_cuda_compiler_version12.9python3.10.____cpython.yaml new file mode 100644 index 0000000..0ade8b1 --- /dev/null +++ b/.ci_support/linux_aarch64_cuda_compiler_version12.9python3.10.____cpython.yaml @@ -0,0 +1,37 @@ +c_compiler: +- gcc +c_compiler_version: +- '14' +c_stdlib: +- sysroot +c_stdlib_version: +- '2.17' +cdt_name: +- conda +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 +pin_run_as_build: + python: + min_pin: x.x + max_pin: x.x +python: +- 3.10.* *_cpython +target_platform: +- linux-aarch64 +zip_keys: +- - c_compiler_version + - cxx_compiler_version + - c_stdlib_version + - cuda_compiler_version diff --git a/.ci_support/linux_aarch64_cuda_compiler_version12.9python3.11.____cpython.yaml b/.ci_support/linux_aarch64_cuda_compiler_version12.9python3.11.____cpython.yaml new file mode 100644 index 0000000..cb92cba --- /dev/null +++ b/.ci_support/linux_aarch64_cuda_compiler_version12.9python3.11.____cpython.yaml @@ -0,0 +1,37 @@ +c_compiler: +- gcc +c_compiler_version: +- '14' +c_stdlib: +- sysroot +c_stdlib_version: +- '2.17' +cdt_name: +- conda +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 +pin_run_as_build: + python: + min_pin: x.x + max_pin: x.x +python: +- 3.11.* *_cpython +target_platform: +- linux-aarch64 +zip_keys: +- - c_compiler_version + - cxx_compiler_version + - c_stdlib_version + - cuda_compiler_version diff --git a/.ci_support/linux_aarch64_cuda_compiler_version12.9python3.12.____cpython.yaml b/.ci_support/linux_aarch64_cuda_compiler_version12.9python3.12.____cpython.yaml new file mode 100644 index 0000000..c8a04f8 --- /dev/null +++ b/.ci_support/linux_aarch64_cuda_compiler_version12.9python3.12.____cpython.yaml @@ -0,0 +1,37 @@ +c_compiler: +- gcc +c_compiler_version: +- '14' +c_stdlib: +- sysroot +c_stdlib_version: +- '2.17' +cdt_name: +- conda +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 +pin_run_as_build: + python: + min_pin: x.x + max_pin: x.x +python: +- 3.12.* *_cpython +target_platform: +- linux-aarch64 +zip_keys: +- - c_compiler_version + - cxx_compiler_version + - c_stdlib_version + - cuda_compiler_version diff --git a/.ci_support/linux_aarch64_cuda_compiler_version12.9python3.13.____cp313.yaml b/.ci_support/linux_aarch64_cuda_compiler_version12.9python3.13.____cp313.yaml new file mode 100644 index 0000000..1a0db03 --- /dev/null +++ b/.ci_support/linux_aarch64_cuda_compiler_version12.9python3.13.____cp313.yaml @@ -0,0 +1,37 @@ +c_compiler: +- gcc +c_compiler_version: +- '14' +c_stdlib: +- sysroot +c_stdlib_version: +- '2.17' +cdt_name: +- conda +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 +pin_run_as_build: + python: + min_pin: x.x + max_pin: x.x +python: +- 3.13.* *_cp313 +target_platform: +- linux-aarch64 +zip_keys: +- - c_compiler_version + - cxx_compiler_version + - c_stdlib_version + - cuda_compiler_version diff --git a/.ci_support/migrations/cuda129.yaml b/.ci_support/migrations/cuda129.yaml new file mode 100644 index 0000000..f8b53b3 --- /dev/null +++ b/.ci_support/migrations/cuda129.yaml @@ -0,0 +1,61 @@ +migrator_ts: 1738229377 +__migrator: + kind: + version + migration_number: + 1 + build_number: + 1 + paused: false + override_cbc_keys: + - cuda_compiler_stub + check_solvable: false + primary_key: cuda_compiler_version + ordering: + cuda_compiler_version: + - 12.4 + - 12.6 + - 12.8 + - None + - 12.9 + - 13.0 + # to allow manual opt-in for CUDA 11.8, see + # https://github.com/conda-forge/conda-forge-pinning-feedstock/pull/7472 + # must be last due to how cuda_compiler ordering in that migrator works + - 11.8 + commit_message: | + Upgrade to CUDA 12.9 + + CUDA 12.8 added support for architectures `sm_100`, `sm_101` and `sm_120`, + while CUDA 12.9 further added `sm_103` and `sm_121`. To build for these, + maintainers will need to modify their existing list of specified architectures + (e.g. `CMAKE_CUDA_ARCHITECTURES`, `TORCH_CUDA_ARCH_LIST`, etc.) + for their package. A good balance between broad support and storage + footprint (resp. compilation time) is to add `sm_100` and `sm_120`. + + Since CUDA 12.8, the conda-forge nvcc package now sets `CUDAARCHS` and + `TORCH_CUDA_ARCH_LIST` in its activation script to a string containing all + of the supported real architectures plus the virtual architecture of the + latest. Recipes for packages who use these variables to control their build + but do not want to build for all supported architectures will need to override + these variables in their build script. + + ref: https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#new-features + +cuda_compiler_version: # [((linux and (x86_64 or aarch64)) or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"] + - 12.9 # [((linux and (x86_64 or aarch64)) or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"] + +cuda_compiler_version_min: # [((linux and (x86_64 or aarch64)) or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"] + - 12.9 # [((linux and (x86_64 or aarch64)) or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"] + +c_stdlib_version: # [(linux and (x86_64 or aarch64)) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"] + - 2.17 # [(linux and (x86_64 or aarch64)) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"] + +c_compiler_version: # [(linux and (x86_64 or aarch64)) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"] + - 14 # [(linux and (x86_64 or aarch64)) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"] + +cxx_compiler_version: # [(linux and (x86_64 or aarch64)) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"] + - 14 # [(linux and (x86_64 or aarch64)) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"] + +fortran_compiler_version: # [(linux and (x86_64 or aarch64)) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"] + - 14 # [(linux and (x86_64 or aarch64)) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"] diff --git a/README.md b/README.md index dc77025..8429c86 100644 --- a/README.md +++ b/README.md @@ -27,31 +27,59 @@ Current build status - + - + - + - + + + + + + + + + + + + + diff --git a/conda-forge.yml b/conda-forge.yml index 1ba88eb..b527956 100644 --- a/conda-forge.yml +++ b/conda-forge.yml @@ -1,5 +1,7 @@ azure: free_disk_space: true +build_platform: + linux_aarch64: linux_64 github: branch_name: main tooling_branch_name: main diff --git a/recipe/recipe.yaml b/recipe/recipe.yaml index c17eef3..f341d1a 100644 --- a/recipe/recipe.yaml +++ b/recipe/recipe.yaml @@ -13,7 +13,7 @@ source: sha256: 1e072841619af9c171709b2c6ff17daa0d9bf7caf0fcb083f5c11754b63c6cdb build: - number: 1 + number: 2 script: env: MAMBA_FORCE_BUILD: "TRUE" @@ -31,9 +31,14 @@ requirements: - ${{ compiler('cxx') }} - ${{ compiler('cuda') }} - ${{ stdlib('c') }} + - ninja + - if: build_platform != target_platform + then: + - python + - cross-python_${{ target_platform }} + - pytorch host: - cuda-version ==${{ cuda_compiler_version }} - - ninja - packaging - pip - python @@ -46,7 +51,7 @@ requirements: - einops - packaging - python - - pytorch + - pytorch * cuda* - setuptools >=61.0.0 - transformers
VariantStatus
linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.10.____cpythonlinux_64_cuda_compiler_version12.9python3.10.____cpython - variant + variant
linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.11.____cpythonlinux_64_cuda_compiler_version12.9python3.11.____cpython - variant + variant
linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.12.____cpythonlinux_64_cuda_compiler_version12.9python3.12.____cpython - variant + variant
linux_64_c_compiler_version13cuda_compiler_version12.6cxx_compiler_version13python3.13.____cp313linux_64_cuda_compiler_version12.9python3.13.____cp313 - variant + variant + +
linux_aarch64_cuda_compiler_version12.9python3.10.____cpython + + variant + +
linux_aarch64_cuda_compiler_version12.9python3.11.____cpython + + variant + +
linux_aarch64_cuda_compiler_version12.9python3.12.____cpython + + variant + +
linux_aarch64_cuda_compiler_version12.9python3.13.____cp313 + + variant