POTRF: register the matrix memory with PaRSEC

Signed-off-by: Joseph Schuchart <[email protected]>
TESSEorg · Feb 15, 2025 · 9dd9a36 · 9dd9a36
1 parent c5cc75b
commit 9dd9a36
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/examples/potrf/testing_dpotrf.cc b/examples/potrf/testing_dpotrf.cc
@@ -108,6 +108,16 @@ int main(int argc, char **argv)
   dcA.mat = parsec_data_allocate((size_t)dcA.super.nb_local_tiles *
                                  (size_t)dcA.super.bsiz *
                                  (size_t)parsec_datadist_getsizeoftype(dcA.super.mtype));
+
+  /* would be nice to have proper abstractions for this */
+  parsec_data_collection_t *o = &(dcA.super.super);
+  for (int devid = 1; devid < parsec_nb_devices; ++devid) {
+    auto* device = parsec_mca_device_get(devid);
+    if (device->memory_register) {
+      o->register_memory(o, device); // TODO: check device IDs
+    }
+  }
+
   parsec_data_collection_set_key((parsec_data_collection_t*)&dcA, (char*)"Matrix A");
 
   if(!check) {