Merge pull request #7931 from dmitrygx/topic/ucp/worker_addr_am_only_…

…v1_12 UCP/WIREUP: Fix parsing AM_ONLY flag from the header of a worker address [v1.12.x]
openucx · Feb 12, 2022 · a6a787d · a6a787d
2 parents 47f786e + ecbcb40
commit a6a787d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/NEWS b/NEWS
@@ -7,6 +7,10 @@
 ##
 #
 
+## 1.12.1-rc2 (TBD)
+### Bugfixes
+* Fixed parsing header flags of worker address
+
 ## 1.12.1-rc1 (February 9, 2022)
 ### Bugfixes
 * Fixed memory hooks for Cuda 11.5

diff --git a/src/ucp/wireup/address.c b/src/ucp/wireup/address.c
@@ -1088,9 +1088,11 @@ uint64_t ucp_address_get_client_id(const void *address)
 
 uint8_t ucp_address_is_am_only(const void *address)
 {
-    const uint8_t address_header = *(const uint8_t *)address;
+    uint8_t addr_flags;
+    ucp_object_version_t addr_version;
 
-    return address_header & UCP_ADDRESS_HEADER_FLAG_AM_ONLY;
+    ucp_address_unpack_header(address, &addr_version, &addr_flags);
+    return addr_flags & UCP_ADDRESS_HEADER_FLAG_AM_ONLY;
 }
 
 static ucs_status_t