File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change @@ -791,23 +791,20 @@ pi_result cuda_piDevicePartition(
791
791
pi_result cuda_piextDeviceSelectBinary (pi_device device,
792
792
pi_device_binary *binaries,
793
793
pi_uint32 num_binaries,
794
- pi_device_binary *selected_binary) {
794
+ pi_uint32 *selected_binary) {
795
795
if (!binaries) {
796
796
cl::sycl::detail::pi::die (" No list of device images provided" );
797
797
}
798
798
if (num_binaries < 1 ) {
799
799
cl::sycl::detail::pi::die (" No binary images in the list" );
800
800
}
801
- if (!selected_binary) {
802
- cl::sycl::detail::pi::die (" No storage for device binary provided" );
803
- }
804
801
805
802
// Look for an image for the NVPTX64 target, and return the first one that is
806
803
// found
807
804
for (pi_uint32 i = 0 ; i < num_binaries; i++) {
808
805
if (strcmp (binaries[i]->DeviceTargetSpec ,
809
806
PI_DEVICE_BINARY_TARGET_NVPTX64) == 0 ) {
810
- *selected_binary = binaries[i] ;
807
+ *selected_binary = i ;
811
808
return PI_SUCCESS;
812
809
}
813
810
}
You can’t perform that action at this time.
0 commit comments