From f9caef6908ecedc7ddb6a0286ac41ceab1ebe260 Mon Sep 17 00:00:00 2001 From: Yiltan Date: Wed, 26 Nov 2025 10:31:23 -0500 Subject: [PATCH] Add rocshmem_int64_p (#335) [ROCm/rocshmem commit: d9e2890222c2a31a9c9b332afd349d308656b945] --- projects/rocshmem/docs/api/rma.rst | 2 +- projects/rocshmem/include/rocshmem/rocshmem_RMA.hpp | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/projects/rocshmem/docs/api/rma.rst b/projects/rocshmem/docs/api/rma.rst index 05570f6501..2ae956fa40 100644 --- a/projects/rocshmem/docs/api/rma.rst +++ b/projects/rocshmem/docs/api/rma.rst @@ -218,7 +218,7 @@ The following table lists the supported RMA data types: - No * - int64_t - int64 - - No + - Yes * - uint8_t - uint8 - No diff --git a/projects/rocshmem/include/rocshmem/rocshmem_RMA.hpp b/projects/rocshmem/include/rocshmem/rocshmem_RMA.hpp index 2be0148cb9..8d627ad633 100644 --- a/projects/rocshmem/include/rocshmem/rocshmem_RMA.hpp +++ b/projects/rocshmem/include/rocshmem/rocshmem_RMA.hpp @@ -405,6 +405,11 @@ __host__ void rocshmem_ctx_ulonglong_p( __host__ void rocshmem_ulonglong_p( unsigned long long *dest, unsigned long long value, int pe); +__device__ ATTR_NO_INLINE void rocshmem_ctx_int64_p( + rocshmem_ctx_t ctx, unsigned long long *dest, unsigned long long value, + int pe); +__device__ ATTR_NO_INLINE void rocshmem_int64_p( + unsigned long long *dest, unsigned long long value, int pe); /** * @name SHMEM_GET