Merge pull request #72 from fjarri/ci

Add POCL-based CI
fjarri · Aug 1, 2024 · 9431c6e · 9431c6e
2 parents 9df7cb9 + dc0ee65
commit 9431c6e
Show file tree

Hide file tree

Showing 7 changed files with 184 additions and 3 deletions.
diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
@@ -0,0 +1,30 @@
+name: Documentation
+
+on:
+  push:
+    branches: [ master ]
+  pull_request:
+    branches: [ master ]
+
+jobs:
+  sphinx-build:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        # synchronized with `.readthedocs.yml`
+        python-version: ["3.10"]
+
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v2
+      with:
+        python-version: ${{ matrix.python-version }}
+    - name: Install PDM
+      run: curl -sSL https://raw.githubusercontent.com/pdm-project/pdm/main/install-pdm.py | python3
+    - name: Install dependencies
+      run: |
+        pdm sync -G docs
+    - name: Run sphinx-build
+      run: |
+        pdm run sphinx-build -n docs build/sphinx
diff --git a/.github/workflows/lints.yml b/.github/workflows/lints.yml
@@ -0,0 +1,38 @@
+name: Lints
+
+on:
+  push:
+    branches: [ master ]
+  pull_request:
+    branches: [ master ]
+
+jobs:
+  ruff:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10"]
+    steps:
+      - uses: actions/checkout@v3
+      - uses: chartboost/ruff-action@v1
+
+  mypy:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10"]
+
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v2
+      with:
+        python-version: ${{ matrix.python-version }}
+    - name: Install PDM
+      run: curl -sSL https://raw.githubusercontent.com/pdm-project/pdm/main/install-pdm.py | python3
+    - name: Install dependencies
+      run: |
+        pdm sync -G lint
+    - name: Run mypy
+      run: |
+        pdm run mypy examples reikna
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -0,0 +1,64 @@
+# This workflow will install Python dependencies, run tests and lint with a variety of Python versions
+# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions
+
+name: Tests
+
+on:
+  push:
+    branches: [ master ]
+  pull_request:
+    branches: [ master ]
+
+jobs:
+
+  test-pocl:
+
+    runs-on: ubuntu-24.04
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.10"]
+
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v2
+      with:
+        python-version: ${{ matrix.python-version }}
+    - name: Install system dependencies
+      run: |
+        sudo apt-get install libpocl2
+    - name: Install PDM
+      run: curl -sSL https://raw.githubusercontent.com/pdm-project/pdm/main/install-pdm.py | python3
+    - name: Install dependencies
+      run: |
+        pdm sync -G tests,pyopencl
+    # Skip performance tests, the test suite is slow enough as it is
+    - name: Test with pytest
+      run: |
+        env POCL_DEVICES="cpu-minimal" pdm run py.test --cov=reikna --cov-report=xml tests -m "not perf"
+    - name: Run demo_fftn_with_transpose
+      run: |
+        env POCL_DEVICES="cpu-minimal" pdm run examples/demo_fftn_with_transpose.py
+    - name: Run demo_fftshift_transformation
+      run: |
+        env POCL_DEVICES="cpu-minimal" pdm run examples/demo_fftshift_transformation.py
+    - name: Run demo_real_to_complex_fft
+      run: |
+        env POCL_DEVICES="cpu-minimal" pdm run examples/demo_real_to_complex_fft.py
+    - name: Run demo_specgram
+      run: |
+        pdm add matplotlib
+        env POCL_DEVICES="cpu-minimal" pdm run examples/demo_specgram.py
+    - name: Run demo_specialized_fft
+      run: |
+        env POCL_DEVICES="cpu-minimal" pdm run examples/demo_specialized_fft.py
+    - name: Run demo_struct_reduce
+      run: |
+        env POCL_DEVICES="cpu-minimal" pdm run examples/demo_struct_reduce.py
+    - name: Upload coverage
+      if: matrix.python-version == '3.10'
+      run: |
+        curl -Os https://uploader.codecov.io/latest/linux/codecov
+        chmod +x codecov
+        ./codecov
diff --git a/README.md b/README.md
@@ -1,5 +1,16 @@
 # Reikna, the pure Python GPGPU library
 
+[![pypi package][pypi-image]][pypi-link] ![License][pypi-license-image] [![Docs][rtd-image]][rtd-link] [![Coverage][cov-image]][cov-link]
+
+[pypi-image]: https://img.shields.io/pypi/v/reikna
+[pypi-link]: https://pypi.org/project/reikna/
+[pypi-license-image]: https://img.shields.io/pypi/l/reikna
+[rtd-image]: https://readthedocs.org/projects/reikna/badge/?version=latest
+[rtd-link]: https://reikna.readthedocs.io/en/latest/
+[cov-image]: https://codecov.io/gh/fjarri/reikna/branch/master/graph/badge.svg
+[cov-link]: https://codecov.io/gh/fjarri/reikna
+
+
 `Reikna` is a library containing various GPU algorithms built on top of [PyCUDA](http://documen.tician.de/pycuda) and [PyOpenCL](http://documen.tician.de/pyopencl).
 The main design goals are:
 

diff --git a/examples/demo_specgram.py b/examples/demo_specgram.py
@@ -170,8 +170,8 @@ def __init__(self, x, NFFT=256, noverlap=128, pad_to=None, window=hanning_window
         rolling_frame_trf = rolling_frame(x, NFFT, noverlap, pad_to)
 
         complex_dtype = dtypes.complex_for(x.dtype)
-        fft_arr = Type(complex_dtype, rolling_frame_trf.output.shape)
-        real_fft_arr = Type(x.dtype, rolling_frame_trf.output.shape)
+        fft_arr = Type.array(complex_dtype, rolling_frame_trf.output.shape)
+        real_fft_arr = Type.array(x.dtype, rolling_frame_trf.output.shape)
 
         window_trf = window(real_fft_arr, NFFT)
         broadcast_zero_trf = transformations.broadcast_const(real_fft_arr, 0)

diff --git a/pdm.lock b/pdm.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -28,6 +28,7 @@ tests = [
     "pytest>=6",
     "pytest-cov",
     "pytest-grunnur>=0.3",
+    "scipy",
 ]
 docs = [
     "sphinx>=4",