Skip to content

Commit b332cbe

Browse files
isVoidisVoid
committed
Merge branch 'main' of github.com:NVIDIA/numbast into bump_numba-cuda_ver
2 parents 96ffc67 + e501c89 commit b332cbe

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

69 files changed

+2704
-291
lines changed

‎.coderabbit.yaml‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
# SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# SPDX-License-Identifier: Apache-2.0
3+
reviews:
4+
profile: assertive # assertive / chill profile
5+
auto_review:
6+
enabled: true # Enable auto-review for this repository

‎.github/actions/docs-build/action.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ runs:
1919
python -m pip install --upgrade pip
2020
# Build and install Numbast from Source
2121
ast_canopy/build.sh
22-
python -m pip install numbast/
22+
python -m pip install --pre numbast/
2323
# Minimum dependencies for Numba-CUDA to run
2424
python -m pip install cuda-python cuda-core
2525
# Install Sphinx dependencies

‎.github/workflows/build-docs.yml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@ jobs:
6767
uses: actions/checkout@v4
6868
with:
6969
persist-credentials: false
70+
fetch-depth: 0
7071
- name: Build docs
7172
if: ${{ !inputs.remove_pr_preview }}
7273
uses: ./.github/actions/docs-build

‎.github/workflows/ci.yaml‎

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# SPDX-License-Identifier: Apache-2.0
3+
name: CI
4+
on:
5+
push:
6+
branches:
7+
- "main"
8+
jobs:
9+
build-conda:
10+
name: Build Conda Python
11+
secrets: inherit
12+
uses: ./.github/workflows/conda-python-build.yaml
13+
with:
14+
build_type: branch
15+
test-conda:
16+
name: Test Conda Python
17+
needs: build-conda
18+
secrets: inherit
19+
uses: ./.github/workflows/conda-python-test.yaml
20+
with:
21+
build_type: branch
22+
build-wheels:
23+
name: Build Wheels
24+
secrets: inherit
25+
uses: ./.github/workflows/wheels-build.yaml
26+
with:
27+
build_type: branch
28+
test-wheels:
29+
name: Test Wheels
30+
needs: build-wheels
31+
secrets: inherit
32+
uses: ./.github/workflows/wheels-test.yaml
33+
with:
34+
build_type: branch

‎.github/workflows/conda-python-build.yaml‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -61,15 +61,15 @@ jobs:
6161
#
6262
export MATRIX="
6363
# amd64
64-
- { ARCH: 'amd64', PY_VER: '3.10', CUDA_VER: '12.8.0', LINUX_VER: 'rockylinux8' }
65-
- { ARCH: 'amd64', PY_VER: '3.11', CUDA_VER: '12.8.0', LINUX_VER: 'rockylinux8' }
66-
- { ARCH: 'amd64', PY_VER: '3.12', CUDA_VER: '12.8.0', LINUX_VER: 'rockylinux8' }
67-
- { ARCH: 'amd64', PY_VER: '3.13', CUDA_VER: '12.8.0', LINUX_VER: 'rockylinux8' }
64+
- { ARCH: 'amd64', PY_VER: '3.10', CUDA_VER: '12.9.1', LINUX_VER: 'rockylinux8' }
65+
- { ARCH: 'amd64', PY_VER: '3.11', CUDA_VER: '12.9.1', LINUX_VER: 'rockylinux8' }
66+
- { ARCH: 'amd64', PY_VER: '3.12', CUDA_VER: '12.9.1', LINUX_VER: 'rockylinux8' }
67+
- { ARCH: 'amd64', PY_VER: '3.13', CUDA_VER: '12.9.1', LINUX_VER: 'rockylinux8' }
6868
# arm64
69-
- { ARCH: 'arm64', PY_VER: '3.10', CUDA_VER: '12.8.0', LINUX_VER: 'rockylinux8' }
70-
- { ARCH: 'arm64', PY_VER: '3.11', CUDA_VER: '12.8.0', LINUX_VER: 'rockylinux8' }
71-
- { ARCH: 'arm64', PY_VER: '3.12', CUDA_VER: '12.8.0', LINUX_VER: 'rockylinux8' }
72-
- { ARCH: 'arm64', PY_VER: '3.13', CUDA_VER: '12.8.0', LINUX_VER: 'rockylinux8' }
69+
- { ARCH: 'arm64', PY_VER: '3.10', CUDA_VER: '12.9.1', LINUX_VER: 'rockylinux8' }
70+
- { ARCH: 'arm64', PY_VER: '3.11', CUDA_VER: '12.9.1', LINUX_VER: 'rockylinux8' }
71+
- { ARCH: 'arm64', PY_VER: '3.12', CUDA_VER: '12.9.1', LINUX_VER: 'rockylinux8' }
72+
- { ARCH: 'arm64', PY_VER: '3.13', CUDA_VER: '12.9.1', LINUX_VER: 'rockylinux8' }
7373
"
7474
7575
MATRIX="$(

‎.github/workflows/conda-python-test.yaml‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -69,21 +69,21 @@ jobs:
6969
pull-request:
7070
# amd64
7171
- { ARCH: 'amd64', PY_VER: '3.10', CUDA_VER: '12.8.0', LINUX_VER: 'rockylinux8', GPU: 'l4', DRIVER: 'earliest' }
72-
- { ARCH: 'amd64', PY_VER: '3.12', CUDA_VER: '12.8.0', LINUX_VER: 'ubuntu24.04', GPU: 'l4', DRIVER: 'latest' }
72+
- { ARCH: 'amd64', PY_VER: '3.12', CUDA_VER: '12.9.1', LINUX_VER: 'ubuntu24.04', GPU: 'l4', DRIVER: 'latest' }
7373
# arm64
7474
- { ARCH: 'arm64', PY_VER: '3.11', CUDA_VER: '12.8.0', LINUX_VER: 'ubuntu22.04', GPU: 'a100', DRIVER: 'latest' }
75-
- { ARCH: 'arm64', PY_VER: '3.11', CUDA_VER: '13.0.1', LINUX_VER: 'rockylinux8', GPU: 'l4', DRIVER: 'latest' }
75+
- { ARCH: 'arm64', PY_VER: '3.13', CUDA_VER: '13.0.1', LINUX_VER: 'rockylinux8', GPU: 'l4', DRIVER: 'latest' }
7676
nightly:
7777
# amd64
78-
- { ARCH: 'amd64', PY_VER: '3.10', CUDA_VER: '11.4.3', LINUX_VER: 'rockylinux8', GPU: 'l4', DRIVER: 'earliest' }
79-
- { ARCH: 'amd64', PY_VER: '3.11', CUDA_VER: '11.4.3', LINUX_VER: 'ubuntu22.04', GPU: 'l4', DRIVER: 'latest' }
80-
- { ARCH: 'amd64', PY_VER: '3.12', CUDA_VER: '11.8.0', LINUX_VER: 'ubuntu24.04', GPU: 'l4', DRIVER: 'latest' }
81-
- { ARCH: 'amd64', PY_VER: '3.13', CUDA_VER: '12.0.1', LINUX_VER: 'rockylinux8', GPU: 'l4', DRIVER: 'latest' }
78+
- { ARCH: 'amd64', PY_VER: '3.10', CUDA_VER: '12.8.0', LINUX_VER: 'rockylinux8', GPU: 'l4', DRIVER: 'earliest' }
79+
- { ARCH: 'amd64', PY_VER: '3.11', CUDA_VER: '12.9.1', LINUX_VER: 'ubuntu22.04', GPU: 'l4', DRIVER: 'latest' }
80+
- { ARCH: 'amd64', PY_VER: '3.12', CUDA_VER: '13.0.1', LINUX_VER: 'ubuntu24.04', GPU: 'l4', DRIVER: 'latest' }
81+
- { ARCH: 'amd64', PY_VER: '3.13', CUDA_VER: '12.9.1', LINUX_VER: 'rockylinux8', GPU: 'l4', DRIVER: 'latest' }
8282
# arm64
83-
- { ARCH: 'arm64', PY_VER: '3.10', CUDA_VER: '11.4.3', LINUX_VER: 'ubuntu22.04', GPU: 'a100', DRIVER: 'latest' }
84-
- { ARCH: 'arm64', PY_VER: '3.11', CUDA_VER: '12.0.1', LINUX_VER: 'ubuntu24.04', GPU: 'a100', DRIVER: 'latest' }
85-
- { ARCH: 'arm64', PY_VER: '3.12', CUDA_VER: '11.8.0', LINUX_VER: 'rockylinux8', GPU: 'a100', DRIVER: 'latest' }
86-
- { ARCH: 'arm64', PY_VER: '3.13', CUDA_VER: '12.2.2', LINUX_VER: 'ubuntu24.04', GPU: 'a100', DRIVER: 'latest' }
83+
- { ARCH: 'arm64', PY_VER: '3.10', CUDA_VER: '12.8.0', LINUX_VER: 'ubuntu22.04', GPU: 'a100', DRIVER: 'latest' }
84+
- { ARCH: 'arm64', PY_VER: '3.11', CUDA_VER: '12.9.1', LINUX_VER: 'ubuntu24.04', GPU: 'l4', DRIVER: 'latest' }
85+
- { ARCH: 'arm64', PY_VER: '3.12', CUDA_VER: '13.0.1', LINUX_VER: 'rockylinux8', GPU: 'a100', DRIVER: 'latest' }
86+
- { ARCH: 'arm64', PY_VER: '3.13', CUDA_VER: '12.9.1', LINUX_VER: 'ubuntu24.04', GPU: 'l4', DRIVER: 'latest' }
8787
"
8888
8989
# Use the nightly matrix for branch tests

‎.github/workflows/publish.yaml‎

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,3 +17,24 @@ jobs:
1717
build_type: release
1818
script: "ci/build_conda_python.sh"
1919
upload_to_conda: true
20+
build-wheels:
21+
uses: ./.github/workflows/wheels-build.yaml
22+
secrets: inherit
23+
with:
24+
build_type: release
25+
upload-to-pypi:
26+
permissions:
27+
id-token: write
28+
needs: build-wheels
29+
runs-on: ubuntu-latest
30+
steps:
31+
- name: Download all wheels artifacts
32+
uses: actions/download-artifact@v4
33+
with:
34+
pattern: pypi-wheels-*
35+
path: dist/
36+
merge-multiple: true
37+
- name: Publish packages to PyPi
38+
uses: pypa/gh-action-pypi-publish@ed0c53931b1dc9bd32cbe73a98c7f6766f8a527e # v1.13.0
39+
with:
40+
verbose: true

‎.github/workflows/wheels-build.yaml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ jobs:
131131
sccache --show-adv-stats
132132
env:
133133
GH_TOKEN: ${{ github.token }}
134-
- name: Upload PyPi wheels
134+
- name: Upload PyPi wheels artifact
135135
uses: actions/upload-artifact@v4
136136
with:
137137
name: pypi-wheels-${{ matrix.ARCH }}

‎VERSION‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
0.5.4
1+
0.5.5

‎ast_canopy/CMakeLists.txt‎

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,17 +3,22 @@
33

44
cmake_minimum_required(VERSION 3.29)
55

6+
file(READ "${CMAKE_CURRENT_SOURCE_DIR}/VERSION" _version_raw)
7+
string(STRIP ${_version_raw} _version)
8+
69
# Oddly, C language is needed here to build in PyTorch image on aarch64
710
# machine
811

912
project(${SKBUILD_PROJECT_NAME}
1013
VERSION ${SKBUILD_PROJECT_VERSION}
1114
LANGUAGES C CXX)
1215

16+
# Prevent C++ modules from being scanned (not used in this project).
17+
set(CMAKE_CXX_SCAN_FOR_MODULES OFF)
1318

1419
find_package(Python REQUIRED COMPONENTS Interpreter Development.Module)
1520
find_package(pybind11 CONFIG REQUIRED)
16-
find_package(astcanopy ${SKBUILD_PROJECT_VERSION} REQUIRED)
21+
find_package(astcanopy ${_version} REQUIRED)
1722

1823
python_add_library(pylibastcanopy MODULE ast_canopy/pylibastcanopy.cpp WITH_SOABI)
1924

0 commit comments

Comments
 (0)