External heap mrfd for fetch atomic

Sandia-OpenSHMEM · Apr 12, 2024 · 4255c12 · 4255c12
1 parent 5989975
commit 4255c12
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/transport_ofi.h b/src/transport_ofi.h
@@ -1304,9 +1304,17 @@ void shmem_transport_fetch_atomic(shmem_transport_ctx_t* ctx, void *target,
         ret = fi_fetch_atomic(ctx->ep,
                               source,
                               1,
+#ifdef USE_FI_HMEM
+                              &shmem_transport_ofi_external_heap_mrfd,
+#else
                               NULL,
+#endif
                               dest,
+#ifdef USE_FI_HMEM
+                              &shmem_transport_ofi_external_heap_mrfd,
+#else
                               NULL,
+#endif
                               GET_DEST(dst),
                               (uint64_t) addr,
                               key,