Skip to content

CI: add seperate workflows to build & import pykokkos on linux and macos #313

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
kennykos wants to merge 10 commits into
base: main
Choose a base branch
from
Open

CI: add seperate workflows to build & import pykokkos on linux and macos #313

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
4f41122
CI: build/import for linux & macos
Nov 17, 2025
8d26fdb
OS build for *only* py3.13
Nov 17, 2025
b0a64dc
Only 1 view rank needed for OS builds
Nov 17, 2025
0b6c763
Merge branch 'main' into ci-build-macos
kennykos Dec 19, 2025
bdd874f
Bring in changes from pykokkos-base/82
kennykos Dec 19, 2025
a401fbf
Install pykokkos/base, not pykokkos-base
kennykos Dec 19, 2025
b2dc3ec
Add arm64 runner
Dec 19, 2025
4d4425b
Change MACOS_RPATH -> INSTALL_RPATH
Dec 19, 2025
3fc036e
Run hello world script for OS builds
Dec 19, 2025
9012574
change base installation
IvanGrigorik Dec 19, 2025
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
39 changes: 39 additions & 0 deletions .github/workflows/build_linux.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
name: Linux

on:
push:
branches:
- main
pull_request:
branches:
- main

jobs:
test_pykokkos:
strategy:
matrix:
platform: [ubuntu-latest, ubuntu-24.04-arm]
python-version: ["3.13"]
runs-on: ${{ matrix.platform }}
steps:
- uses: actions/checkout@v3
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v4
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
run: |
python -m pip install --upgrade pip
python -m pip install --upgrade numpy mypy==1.0.1 cmake pytest pybind11 scikit-build patchelf
- name: Install pykokkos-base
run: |
python install_base.py install -- -DENABLE_LAYOUTS=ON -DENABLE_MEMORY_TRAITS=OFF -DENABLE_VIEW_RANKS=1
- name: Install pykokkos
run: |
python -m pip install .
- name: mypy check
run: |
mypy pykokkos
- name: import pykokkos
run: |
python -c "import pykokkos"
39 changes: 39 additions & 0 deletions .github/workflows/build_macos.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
name: MacOS

on:
push:
branches:
- main
pull_request:
branches:
- main

jobs:
test_pykokkos:
strategy:
matrix:
platform: [macos-latest]
python-version: ["3.13"]
runs-on: ${{ matrix.platform }}
steps:
- uses: actions/checkout@v3
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v4
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
run: |
python -m pip install --upgrade pip
python -m pip install --upgrade numpy mypy==1.0.1 cmake pytest pybind11 scikit-build patchelf
- name: Install pykokkos-base
run: |
python install_base.py install -- -DENABLE_LAYOUTS=ON -DENABLE_MEMORY_TRAITS=OFF -DENABLE_VIEW_RANKS=1
- name: Install pykokkos
run: |
python -m pip install .
- name: mypy check
run: |
mypy pykokkos
- name: import pykokkos
run: |
python examples/kokkos/01_hello_world.py
3 changes: 1 addition & 2 deletions .github/workflows/main_ci.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,9 +26,8 @@ jobs:
python -m pip install --upgrade pip
python -m pip install --upgrade numpy mypy==1.0.1 cmake pytest pybind11 scikit-build patchelf
- name: Install pykokkos-base
working-directory: base
run: |
python setup.py install -- -DENABLE_LAYOUTS=ON -DENABLE_MEMORY_TRAITS=OFF -DENABLE_VIEW_RANKS=5
python install_base.py install -- -DENABLE_LAYOUTS=ON -DENABLE_MEMORY_TRAITS=OFF -DENABLE_VIEW_RANKS=5
- name: Install pykokkos
run: |
python -m pip install .
Expand Down
2 changes: 2 additions & 0 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# PyKokkos
[![Python Testing](https://github.com/kokkos/pykokkos/actions/workflows/main_ci.yml/badge.svg)](https://github.com/kokkos/pykokkos/actions/workflows/main_ci.yml)
[![Documentation](https://github.com/kokkos/pykokkos/actions/workflows/documentation.yml/badge.svg)](https://github.com/kokkos/pykokkos/actions/workflows/documentation.yml)
[![Linux](https://github.com/kokkos/pykokkos/actions/workflows/build_linux.yml/badge.svg)](https://github.com/kokkos/pykokkos/actions/workflows/build_linux.yml)
[![MacOS](https://github.com/kokkos/pykokkos/actions/workflows/build_macos.yml/badge.svg)](https://github.com/kokkos/pykokkos/actions/workflows/build_macos.yml)

PyKokkos is a framework for writing high-performance Python code
similar to Numba. In contrast to Numba, PyKokkos kernels are primarily
Expand Down
2 changes: 1 addition & 1 deletion base/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,7 @@ SET_TARGET_PROPERTIES(libpykokkos PROPERTIES
# configure the rpath: <RELATIVE>:<CWD>:<FULL>
IF(APPLE)
SET_TARGET_PROPERTIES(libpykokkos PROPERTIES
MACOSX_RPATH "@loader_path/${LIB_RELPATH}:@loader_path:${_Kokkos_LIBDIR}:${CMAKE_INSTALL_RPATH}")
INSTALL_RPATH "@loader_path/${LIB_RELPATH};@loader_path;${_Kokkos_LIBDIR};${CMAKE_INSTALL_RPATH}")
ELSEIF(UNIX)
SET_TARGET_PROPERTIES(libpykokkos PROPERTIES
INSTALL_RPATH "\$ORIGIN/${LIB_RELPATH}:\$ORIGIN:${_Kokkos_LIBDIR}:${CMAKE_INSTALL_RPATH}")
Expand Down
Loading