Feat/custom cuda kernels

.github/workflows/ci.yml

@@ -45,4 +45,4 @@ jobs:
         run: poetry install
 
       - name: Run tests
-        run: poetry run pytest tests/ -v --cov=pygaborstm --cov-report=term-missing
+        run: poetry run pytest tests/ -v -m "not gpu" --cov=pygaborstm --cov-report=term-missing

.gitignore

@@ -45,4 +45,6 @@ ENV/
 logs/
 outputs/
 checkpoints/
-*.log
+*.log
+
+images/

README.md

@@ -107,7 +107,8 @@ PyGaborSTM/
 ```bash
 poetry install                      # Install all dependencies
 poetry run jupyter notebook         # Run notebooks
-poetry run pytest -v                # Run tests
+pytest -m "not gpu"                 # Run all tests exluding GPU kernel tests (used in CI/CD)
+poetry run pytest -v                # Run all tests including GPU kernel tests
 poetry run ruff check --fix .       # lint and fix
 poetry run ruff format .            # format code
 ```