Support to write packets via uint2 (#327)

microsoft · Jul 15, 2024 · 5f9ee27 · 5f9ee27
1 parent c4ca2fb
commit 5f9ee27
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/include/mscclpp/packet_device.hpp b/include/mscclpp/packet_device.hpp
@@ -59,6 +59,11 @@ union alignas(16) LL16Packet {
   /// @param flag The flag to write.
   MSCCLPP_DEVICE_INLINE void write(uint64_t val, uint32_t flag) { write((uint32_t)val, (uint32_t)(val >> 32), flag); }
 
+  /// Write 8 bytes of data to the packet.
+  /// @param val The 8-byte data to write.
+  /// @param flag The flag to write.
+  MSCCLPP_DEVICE_INLINE void write(uint2 val, uint32_t flag) { write(val.x, val.y, flag); }
+
   /// Helper of @ref read().
   /// @param flag The flag to read.
   /// @param data The 8-byte data read.