[SYCL][CUDA] Select only NVPTX64 device binaries

Search through the available binary images and select the first one for
the PI_DEVICE_BINARY_TARGET_NVPTX64 ("nvptx64") target.

Return PI_INVALID_BINARY if no "nvptx64" image is available.

Signed-off-by: Andrea Bocci <[email protected]>

Author: fwyzard

sycl/plugins/cuda/pi_cuda.cpp

@@ -673,10 +673,10 @@ pi_result cuda_piDevicePartition(
   return {};
 }
 
-pi_result cuda_piextDeviceSelectBinary(
-    pi_device device, // TODO: does this need to be context?
-    pi_device_binary *binaries, pi_uint32 num_binaries,
-    pi_device_binary *selected_binary) {
+pi_result cuda_piextDeviceSelectBinary(pi_device device,
+                                       pi_device_binary *binaries,
+                                       pi_uint32 num_binaries,
+                                       pi_device_binary *selected_binary) {
   if (!binaries) {
     cl::sycl::detail::pi::die("No list of device images provided");
   }
@@ -686,8 +686,19 @@ pi_result cuda_piextDeviceSelectBinary(
   if (!selected_binary) {
     cl::sycl::detail::pi::die("No storage for device binary provided");
   }
-  *selected_binary = binaries[0];
-  return PI_SUCCESS;
+
+  // Look for an image for the NVPTX64 target, and return the first one that is
+  // found
+  for (pi_uint32 i = 0; i < num_binaries; i++) {
+    if (strcmp(binaries[i]->DeviceTargetSpec,
+               PI_DEVICE_BINARY_TARGET_NVPTX64) == 0) {
+      *selected_binary = binaries[i];
+      return PI_SUCCESS;
+    }
+  }
+
+  // No image can be loaded for the given device
+  return PI_INVALID_BINARY;
 }
 
 pi_result cuda_piextGetDeviceFunctionPointer(pi_device device,