gpu-datain: pure distribution collections have no data_of/data_of_key

ICLDisco · Oct 11, 2024 · efa8386 · efa8386
1 parent 22254f9
commit efa8386
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/parsec/remote_dep_mpi.c b/parsec/remote_dep_mpi.c
@@ -827,9 +827,11 @@ remote_dep_mpi_retrieve_datatype(parsec_execution_stream_t *eu,
     if (NULL != fct->data_affinity ) {
         parsec_data_ref_t dref;
         fct->data_affinity(newcontext, &dref);
-        parsec_data_t* data = dref.dc->data_of_key(dref.dc, dref.key);
-        output->data.preferred_device = (-1 != data->preferred_device) ?
-                                        data->preferred_device : data->owner_device; 
+        if(NULL != dref.dc->data_of_key) {
+            parsec_data_t* data = dref.dc->data_of_key(dref.dc, dref.key);
+            output->data.preferred_device = (-1 != data->preferred_device) ?
+                                            data->preferred_device : data->owner_device;
+        }
     }
     return PARSEC_ITERATE_CONTINUE;
 }