Initial commit

Author: shwina

.github/workflows/mypy-check.yml

@@ -0,0 +1,38 @@
+name: "Optional mypy check"
+
+on:
+  push:
+    branches:
+      - "pull-request/[0-9]+"
+      - "main"
+    required: false
+
+jobs:
+  mypy:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout ${{ github.event.repository.name }}
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      # WAR: setup-python is not relocatable...
+      # see https://github.com/actions/setup-python/issues/871
+      - name: Set up Python ${{ matrix.python-version }}
+        if: ${{ startsWith(matrix.host-platform, 'linux') }}
+        id: setup-python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+
+      - name: Install dependencies
+        run: |
+          cd cuda_core
+          python -m pip install --upgrade pip
+          pip install -e .
+          pip install mypy
+
+      - name: Run Mypy
+        run: |
+          mypy --no-namespace-packages . || true

cuda_bindings/cuda/py.typed

@@ -109,3 +109,9 @@ exclude = ["cuda/bindings/_version.py"]
   "UP022",
   "E402", # module level import not at top of file
   "F841"] # F841 complains about unused variables, but some assignments have side-effects that could be useful for tests (func calls for example)
+
+
+[[tool.mypy.overrides]]
+module = ["Cython", "llvmlite"]
+ignore_missing_imports = true
+follow_imports = "skip"