OSHMEM/MCA/SPML/UCX: v1.5 implement put_signal and put_signal_nbi

oshmem/mca/spml/ucx/spml_ucx.c

@@ -1644,19 +1644,67 @@ int mca_spml_ucx_put_all_nb(void *dest, const void *source, size_t size, long *c
     return OSHMEM_SUCCESS;
 }
 
-/* This routine is not implemented */
+static inline int mca_spml_ucx_signal_common(shmem_ctx_t ctx,
+                                             uint64_t *sig_addr,
+                                             uint64_t signal,
+                                             int sig_op,
+                                             int dst,
+                                             int blocking)
+{
+    uint64_t dummy_prev, dummy_fetch;
+
+    if (sig_op == SHMEM_SIGNAL_SET) {
+        /* Use atomic swap to set the signal value */
+        if (blocking) {
+            return MCA_ATOMIC_CALL(swap(ctx, (void*)sig_addr, (void*)&dummy_prev,
+                                   signal, sizeof(uint64_t), dst));
+        }
+        return MCA_ATOMIC_CALL(swap_nb(ctx, &dummy_fetch, (void*)sig_addr, (void*)&dummy_prev,
+                                    signal, sizeof(uint64_t), dst));
+    } else if (sig_op == SHMEM_SIGNAL_ADD) {
+        /* Use atomic add to add the signal value */
+        if (blocking) {
+            return MCA_ATOMIC_CALL(add(ctx, (void*)sig_addr, signal,
+                                   sizeof(uint64_t), dst));
+        }
+        return MCA_ATOMIC_CALL(fadd_nb(ctx, &dummy_fetch, (void*)sig_addr, (void*)&dummy_prev,
+                                   signal, sizeof(uint64_t), dst));
+    }
+
+    SPML_UCX_ERROR("Invalid signal operation: %d", sig_op);
+    return OSHMEM_ERR_NOT_IMPLEMENTED;
+}
+
 int mca_spml_ucx_put_signal(shmem_ctx_t ctx, void* dst_addr, size_t size, void*
         src_addr, uint64_t *sig_addr, uint64_t signal, int sig_op, int dst)
 {
-    return OSHMEM_ERR_NOT_IMPLEMENTED;
+    int res;
+
+    res = mca_spml_ucx_put(ctx, dst_addr, size, src_addr, dst);
+    if (OPAL_UNLIKELY(OSHMEM_SUCCESS != res)) {
+        return res;
+    }
+
+    return mca_spml_ucx_signal_common(ctx, sig_addr, signal, sig_op, dst, 1);
 }
 
-/* This routine is not implemented */
 int mca_spml_ucx_put_signal_nb(shmem_ctx_t ctx, void* dst_addr, size_t size,
         void* src_addr, uint64_t *sig_addr, uint64_t signal, int sig_op, int
         dst)
 {
-    return OSHMEM_ERR_NOT_IMPLEMENTED;
+    int res;
+
+    res = mca_spml_ucx_put_nb(ctx, dst_addr, size, src_addr, dst, NULL);
+    if (OPAL_UNLIKELY(OSHMEM_SUCCESS != res)) {
+        return res;
+    }
+
+    res = mca_spml_ucx_fence(ctx);
+    if (OPAL_UNLIKELY(OSHMEM_SUCCESS != res)) {
+        return res;
+    }
+
+    return mca_spml_ucx_signal_common(ctx, sig_addr, signal, sig_op, dst, 0);
 }
 
 /* This routine is not implemented */