Add rocshmem_int64_p (#335)

[ROCm/rocshmem commit: d9e2890222]
2025-11-26 10:31:23 -05:00
parent a40485d292
commit f9caef6908
2 changed files with 6 additions and 1 deletions
@@ -218,7 +218,7 @@ The following table lists the supported RMA data types:
      - No
    * - int64_t
      - int64
-      - No
+      - Yes
    * - uint8_t
      - uint8
      - No
@@ -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