Merge pull request #267 from Cambridge-ICCS/hotfix-cuda-device-index-…

…test Use `get_device_index` method in test
Cambridge-ICCS · Jan 30, 2025 · 46dfa7e · 46dfa7e
2 parents c21040d + b0d8ded
commit 46dfa7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/test/unit/test_tensor_interrogation_cuda.pf b/src/test/unit/test_tensor_interrogation_cuda.pf
@@ -34,7 +34,7 @@ contains
     call torch_tensor_empty(tensor, ndims, tensor_shape, dtype, device_type)
 
     ! Check that torch_tensor_get_device_index can get the device index
-    if (expected /= torch_tensor_get_device_index(tensor)) then
+    if (expected /= tensor%get_device_index()) then
       call torch_tensor_delete(tensor)
       print *, "Error :: torch_tensor_get_device_index returned incorrect CUDA device index"
       stop 999