Array API Implementation (#12)

* Add fixed array-api code

* Add Device support

* Add docs

* Fix bugs

* Add tags

* Update requirements

* Fix missing dependencies.

* Fix CI and Makefile

* Move setup.py and configs to pyproject.toml

Author: roaffix

.github/workflows/ci.yml

@@ -20,116 +20,74 @@ defaults:
     shell: bash -l {0}
 
 jobs:
-  style:
-    name: Style
+  code_style:
+    name: ${{ matrix.check.name }}
     runs-on: ubuntu-latest
-
+    strategy:
+      matrix:
+        check:
+          - name: Code Formatting | black
+            command: black --check .
+          - name: Import Ordering | isort
+            command: isort --check .
+          - name: Linting | flake8
+            command: flake8 .
+          - name: Type Checking | mypy
+            command: mypy . --cache-dir=/dev/null
     steps:
       - uses: actions/checkout@v3
-
+      
       - name: Setup Python
         uses: actions/setup-python@v4
         with:
           python-version: ${{ env.DEFAULT_PYTHON_VERSION }}
 
-      - name: Install requirements
-        run: |
-          grep -E '^black' dev-requirements.txt | xargs pip install
+      - name: Cache Python Environment
+        uses: actions/cache@v2
+        with:
+          path: .venv
+          key: ${{ runner.os }}-pyenv-${{ hashFiles('dev-requirements.txt') }}
+          restore-keys: |
+            ${{ runner.os }}-pyenv-
 
-      - name: Debug info
+      - name: Install Development Requirements
         run: |
-          pip freeze
+          python -m venv .venv
+          source .venv/bin/activate
+          pip install -r dev-requirements.txt
 
-      - name: Run black
+      - name: Run Code Style Check
         run: |
-          black --check .
-
-  checks:
-    name: ${{ matrix.task.name }}
-    runs-on: ${{ matrix.task.runs_on }}
-    timeout-minutes: 30
-    strategy:
-      fail-fast: false
-      matrix:
-        task:
-          - name: Lint
-            runs_on: ubuntu-latest
-            coverage_report: false
-            platform: cpu
-            run: |
-              make flake8
-              make import-sort
-              make typecheck
-
-          - name: CPU Tests
-            runs_on: ubuntu-latest
-            coverage_report: true
-            platform: cpu
-            run: make tests
+          source .venv/bin/activate
+          ${{ matrix.check.command }}
 
+  unit_tests:
+    name: Unit Tests
+    runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v3
-
-      - uses: conda-incubator/setup-miniconda@v2
+      
+      - name: Setup Python
+        uses: actions/setup-python@v4
         with:
-          miniconda-version: "latest"
           python-version: ${{ env.DEFAULT_PYTHON_VERSION }}
 
-      - name: Set build variables
-        run: |
-          # Get the exact Python version to use in the cache key.
-          echo "PYTHON_VERSION=$(python --version)" >> $GITHUB_ENV
-          echo "RUNNER_ARCH=$(uname -m)" >> $GITHUB_ENV
-          # Use week number in cache key so we can refresh the cache weekly.
-          echo "WEEK_NUMBER=$(date +%V)" >> $GITHUB_ENV
-
-      - uses: actions/cache@v3
-        id: virtualenv-cache
+      - name: Cache Python Environment
+        uses: actions/cache@v2
         with:
           path: .venv
-          key: >
-            ${{ env.CACHE_PREFIX }}-${{ env.WEEK_NUMBER }}-${{ runner.os }}-${{ env.RUNNER_ARCH }}-
-            ${{ env.PYTHON_VERSION }}-${{ matrix.task.platform }}-${{ hashFiles('setup.py') }}-
-            ${{ hashFiles('*requirements.txt') }}
-
-      - name: Setup virtual environment (no cache hit)
-        if: steps.virtualenv-cache.outputs.cache-hit != 'true'
-        run: |
-          python${{ env.DEFAULT_PYTHON_VERSION }} -m venv .venv
-          source .venv/bin/activate
-          make install
-
-      - name: Setup virtual environment (cache hit)
-        if: steps.virtualenv-cache.outputs.cache-hit == 'true'
-        run: |
-          source .venv/bin/activate
-          pip install --no-deps -e .[all]
-
-      - name: Debug info
-        run: |
-          source .venv/bin/activate
-          pip freeze
+          key: ${{ runner.os }}-pyenv-${{ hashFiles('requirements.txt') }}
+          restore-keys: |
+            ${{ runner.os }}-pyenv-
 
-      - name: ${{ matrix.task.name }}
+      - name: Install Development Requirements
         run: |
+          python -m venv .venv
           source .venv/bin/activate
-          ${{ matrix.task.run }}
-
-      - name: Prepare coverage report
-        if: matrix.task.coverage_report
-        run: |
-          mkdir coverage
-          mv coverage.xml coverage/
-
-      - name: Save coverage report
-        if: matrix.task.coverage_report
-        uses: actions/upload-artifact@v3
-        with:
-          name: ${{ matrix.task.name }}-coverage
-          path: ./coverage
+          pip install -r requirements.txt
 
-      - name: Clean up
-        if: always()
+      - name: Run Tests
         run: |
           source .venv/bin/activate
-          pip uninstall --yes arrayfire
+          make tests
+        

.isort.cfg

@@ -14,26 +14,13 @@ endif
 version : 
 	@python -c 'from arrayfire.version import VERSION; print(f"ArrayFire Python v{VERSION}")'
 
-.PHONY : install
-install :
-	pip install --upgrade pip
-	pip install pip-tools
-	pip-compile requirements.in -o final_requirements.txt --allow-unsafe --rebuild --verbose
-	pip install -e . -r final_requirements.txt
+# Dev
 
-# Testing
-
-.PHONY : flake8
-flake8 :
-	flake8 arrayfire tests examples
+.PHONY : pre-commit
+pre-commit :
+	black --check . && isort --check . && flake8 . && mypy . --cache-dir=/dev/null
 
-.PHONY : import-sort
-import-sort :
-	isort arrayfire tests examples
-
-.PHONY : typecheck
-typecheck :
-	mypy arrayfire tests examples --cache-dir=/dev/null
+# Testing
 
 .PHONY : tests
 tests :

Makefile

@@ -95,6 +95,7 @@
     "shift",
     "tile",
     "transpose",
+    "lookup",
 ]
 
 from arrayfire.library.array_functions import (
@@ -110,6 +111,7 @@
     isnan,
     iszero,
     join,
+    lookup,
     lower,
     moddims,
     pad,
@@ -591,27 +593,6 @@
 
 from arrayfire.library.statistics import corrcoef, cov, mean, median, stdev, topk, var
 
-# TODO
-# Temp solution. Remove when arrayfire-binary-python-wrapper is finalized
-
-# __all__ += [
-#     "get_active_backend",
-#     "get_available_backends",
-#     "get_backend_count",
-#     "get_backend_id",
-#     "get_device_id",
-#     "set_backend",
-# ]
-
-# from arrayfire.library.unified_api_functions import (
-#     get_active_backend,
-#     get_available_backends,
-#     get_backend_count,
-#     get_backend_id,
-#     get_device_id,
-#     set_backend,
-# )
-
 __all__ += [
     "accum",
     "scan",