Merge pull request #15 from fjarri/multi-device

Add a second virtual device and run examples as a part of POCL test

Author: fjarri

.github/workflows/tests.yml

@@ -62,5 +62,11 @@ jobs:
       run: |
         pdm sync -G tests,pyopencl
     - name: Test with pytest
+      # Create two POCL devices so that we could run multi-device tests
       run: |
-        pdm run py.test --cov=grunnur --cov-report=xml tests
+        env POCL_DEVICES="cpu cpu" pdm run py.test --cov=grunnur --cov-report=xml tests
+    - name: Run examples
+      # Create two POCL devices so that we could run multi-device tests
+      run: |
+        env POCL_DEVICES="cpu cpu" pdm run examples/multi_device_comparison.py
+        env POCL_DEVICES="cpu cpu" pdm run examples/temporary_allocations.py

examples/multi_device_comparison.py

@@ -101,10 +101,10 @@ def test_multi_device(
         assert (a_ref == a_res).all()
 
 
+test_single_device(0, 2**20)
 test_single_device(1, 2**20)
-test_single_device(2, 2**20)
-test_multi_device([1, 2], 2**20)
+test_multi_device([0, 1], 2**20)
 
+test_single_device(0, 2**24, benchmark=True)
 test_single_device(1, 2**24, benchmark=True)
-test_single_device(2, 2**24, benchmark=True)
-test_multi_device([1, 2], 2**24, benchmark=True)
+test_multi_device([0, 1], 2**24, benchmark=True)