Select correct devices when tests run with XOBJECTS_TEST_CONTEXTS

xsuite · Jun 24, 2024 · 87939d5 · 87939d5
1 parent 53d3a5f
commit 87939d5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/xobjects/context_cupy.py b/xobjects/context_cupy.py
@@ -473,6 +473,9 @@ def build_kernels(
 
         return out_kernels
 
+    def __str__(self):
+        return f"{type(self).__name__}:{cupy.cuda.get_device_id()}"
+
     def nparray_to_context_array(self, arr):
         """
         Copies a numpy array to the device memory.

diff --git a/xobjects/context_pyopencl.py b/xobjects/context_pyopencl.py
@@ -219,7 +219,7 @@ def build_kernels(
                 fid.write(specialized_source)
 
         prg = cl.Program(self.context, specialized_source).build(
-            options="-cl-std=CL2.0"
+            options="-cl-std=CL2.0",
         )
 
         out_kernels = {}
@@ -238,6 +238,11 @@ def build_kernels(
 
         return out_kernels
 
+    def __str__(self):
+        platform_id = cl.get_platforms().index(self.platform)
+        device_id = self.platform.get_devices().index(self.device)
+        return f"{type(self).__name__}:{platform_id}.{device_id}"
+
     def nparray_to_context_array(self, arr):
         """
         Copies a numpy array to the device memory.