oshmem: Don't deduplicate rkeys of local node

Author: tvegas1

oshmem/mca/spml/ucx/spml_ucx.c

@@ -255,7 +255,11 @@ int mca_spml_ucx_ctx_mkey_add(mca_spml_ucx_ctx_t *ucx_ctx, int pe, uint32_t segn
             return OSHMEM_ERROR;
         }
 
-        (*ucx_mkey)->rkey = mca_spml_ucx_rkey_store_get(&mca_spml_ucx.rkey_store, rkey);
+        if (!oshmem_proc_on_local_node(pe)) {
+            rkey = mca_spml_ucx_rkey_store_get(&mca_spml_ucx.rkey_store, rkey);
+        }
+
+        (*ucx_mkey)->rkey = rkey;
 
         rc = mca_spml_ucx_ctx_mkey_cache(ucx_ctx, mkey, segno, pe);
         if (OSHMEM_SUCCESS != rc) {
@@ -271,7 +275,11 @@ int mca_spml_ucx_ctx_mkey_del(mca_spml_ucx_ctx_t *ucx_ctx, int pe, uint32_t segn
     ucp_peer_t *ucp_peer;
     int rc;
     ucp_peer = &(ucx_ctx->ucp_peers[pe]);
-    mca_spml_ucx_rkey_store_put(&mca_spml_ucx.rkey_store, ucx_mkey->rkey);
+    if (!oshmem_proc_on_local_node(pe)) {
+        mca_spml_ucx_rkey_store_put(&mca_spml_ucx.rkey_store, ucx_mkey->rkey);
+    } else {
+        ucp_rkey_destroy(ucx_mkey->rkey);
+    }
     ucx_mkey->rkey = NULL;
     rc = mca_spml_ucx_peer_mkey_cache_del(ucp_peer, segno);
     if(OSHMEM_SUCCESS != rc){