UCX common: encapsulate storing of a 64bit value to variable-size pointer

Signed-off-by: Joseph Schuchart <[email protected]>

Author: devreal

opal/mca/common/ucx/common_ucx.h

@@ -135,6 +135,24 @@ uint64_t opal_common_ucx_load_uint64(const void *ptr, size_t size)
     }
 }
 
+/**
+ * Cast and store a uint64_t value to a value of \c size bytes pointed to by \c ptr.
+ */
+static inline
+void opal_common_ucx_store_uint64(uint64_t value, void *ptr, size_t size)
+{
+    if (sizeof(uint8_t) == size) {
+      *(uint8_t*)ptr  = value;
+    } else if (sizeof(uint16_t) == size) {
+      *(uint16_t*)ptr = value;
+    } else if (sizeof(uint32_t) == size) {
+      *(uint32_t*)ptr = value;
+    } else {
+      *(uint64_t*)ptr = value;
+    }
+}
+
+
 static inline
 ucs_status_t opal_common_ucx_request_status(ucs_status_ptr_t request)
 {
@@ -226,13 +244,7 @@ int opal_common_ucx_atomic_cswap(ucp_ep_h ep, uint64_t compare,
                                  uint64_t remote_addr, ucp_rkey_h rkey,
                                  ucp_worker_h worker)
 {
-    if (op_size == sizeof(uint64_t)) {
-        *(uint64_t*)result = value;
-    } else {
-        assert(op_size == sizeof(uint32_t));
-        *(uint32_t*)result = value;
-    }
-
+    opal_common_ucx_store_uint64(value, result, op_size);
     return opal_common_ucx_atomic_fetch(ep, UCP_ATOMIC_FETCH_OP_CSWAP, compare, result,
                                         op_size, remote_addr, rkey, worker);
 }
@@ -244,13 +256,7 @@ ucs_status_ptr_t opal_common_ucx_atomic_cswap_nb(ucp_ep_h ep, uint64_t compare,
                                                  ucp_send_callback_t req_handler,
                                                  ucp_worker_h worker)
 {
-    if (op_size == sizeof(uint64_t)) {
-        *(uint64_t*)result = value;
-    } else {
-        assert(op_size == sizeof(uint32_t));
-        *(uint32_t*)result = value;
-    }
-
+    opal_common_ucx_store_uint64(value, result, op_size);
     return opal_common_ucx_atomic_fetch_nb(ep, UCP_ATOMIC_FETCH_OP_CSWAP, compare, result,
                                            op_size, remote_addr, rkey, req_handler, worker);
 }