device-ir: rename APIs (C/S suffix), add docs, add scalar IR tests

Author: MC952-arch

bindings/ir/flagcx_device_scalar_ir.h

@@ -1,68 +1,124 @@
-flagcxCoopAnyInitBlock
-flagcxCoopAnyInitLanes
-flagcxCoopAnyInitThread
-flagcxCoopAnyInitTileSpan
-flagcxCoopAnyInitWarp
+flagcxCoopAnyInitBlockC
+flagcxCoopAnyInitLanesC
+flagcxCoopAnyInitThreadC
+flagcxCoopAnyInitTileSpanC
+flagcxCoopAnyInitWarpC
 flagcxCoopSizeC
+flagcxCoopSizeExS
+flagcxCoopSizeS
 flagcxCoopSyncC
+flagcxCoopSyncExS
+flagcxCoopSyncS
 flagcxCoopThreadRankC
+flagcxCoopThreadRankExS
+flagcxCoopThreadRankS
 flagcxDataTypeSizeDevice
 flagcxDevCommGetIntraRank
 flagcxDevCommGetIntraSize
 flagcxDevCommGetRank
 flagcxDevCommGetSize
-flagcxDevNetFlush
-flagcxDevNetGet
+flagcxDevNetFlushC
+flagcxDevNetFlushS
+flagcxDevNetGetC
+flagcxDevNetGetFromCommS
+flagcxDevNetGetS
 flagcxDevNetIncreaseSignalShadow
 flagcxDevNetInitC
-flagcxDevNetPut
-flagcxDevNetPutValue
-flagcxDevNetPutValue_RCtrInc
-flagcxDevNetPutValue_RSigAdd
-flagcxDevNetPutValue_RSigInc
-flagcxDevNetPut_LCtrInc
-flagcxDevNetPut_LSigAdd
-flagcxDevNetPut_LSigInc
-flagcxDevNetPut_RCtrInc
-flagcxDevNetPut_RCtrInc_LCtrInc
-flagcxDevNetPut_RCtrInc_LSigAdd
-flagcxDevNetPut_RCtrInc_LSigInc
-flagcxDevNetPut_RSigAdd
-flagcxDevNetPut_RSigAdd_LCtrInc
-flagcxDevNetPut_RSigAdd_LSigAdd
-flagcxDevNetPut_RSigAdd_LSigInc
-flagcxDevNetPut_RSigInc
-flagcxDevNetPut_RSigInc_LCtrInc
-flagcxDevNetPut_RSigInc_LSigAdd
-flagcxDevNetPut_RSigInc_LSigInc
-flagcxDevNetReadCounter
-flagcxDevNetReadSignal
-flagcxDevNetRecv
+flagcxDevNetPutC
+flagcxDevNetPutS
+flagcxDevNetPutS_LCtrInc
+flagcxDevNetPutS_LSigAdd
+flagcxDevNetPutS_LSigInc
+flagcxDevNetPutS_RCtrInc
+flagcxDevNetPutS_RCtrInc_LCtrInc
+flagcxDevNetPutS_RCtrInc_LSigAdd
+flagcxDevNetPutS_RCtrInc_LSigInc
+flagcxDevNetPutS_RSigAdd
+flagcxDevNetPutS_RSigAdd_LCtrInc
+flagcxDevNetPutS_RSigAdd_LSigAdd
+flagcxDevNetPutS_RSigAdd_LSigInc
+flagcxDevNetPutS_RSigInc
+flagcxDevNetPutS_RSigInc_LCtrInc
+flagcxDevNetPutS_RSigInc_LSigAdd
+flagcxDevNetPutS_RSigInc_LSigInc
+flagcxDevNetPutValueC
+flagcxDevNetPutValueS
+flagcxDevNetPutValueS_RCtrInc
+flagcxDevNetPutValueS_RSigAdd
+flagcxDevNetPutValueS_RSigInc
+flagcxDevNetPutValue_RCtrIncC
+flagcxDevNetPutValue_RSigAddC
+flagcxDevNetPutValue_RSigIncC
+flagcxDevNetPut_LCtrIncC
+flagcxDevNetPut_LSigAddC
+flagcxDevNetPut_LSigIncC
+flagcxDevNetPut_RCtrIncC
+flagcxDevNetPut_RCtrInc_LCtrIncC
+flagcxDevNetPut_RCtrInc_LSigAddC
+flagcxDevNetPut_RCtrInc_LSigIncC
+flagcxDevNetPut_RSigAddC
+flagcxDevNetPut_RSigAdd_LCtrIncC
+flagcxDevNetPut_RSigAdd_LSigAddC
+flagcxDevNetPut_RSigAdd_LSigIncC
+flagcxDevNetPut_RSigIncC
+flagcxDevNetPut_RSigInc_LCtrIncC
+flagcxDevNetPut_RSigInc_LSigAddC
+flagcxDevNetPut_RSigInc_LSigIncC
+flagcxDevNetReadCounterC
+flagcxDevNetReadCounterS
+flagcxDevNetReadSignalC
+flagcxDevNetReadSignalS
+flagcxDevNetRecvC
+flagcxDevNetRecvS
 flagcxDevNetResetCounter
 flagcxDevNetResetSignal
-flagcxDevNetSend
-flagcxDevNetSignalCtrInc
-flagcxDevNetSignalSigAdd
-flagcxDevNetSignalSigInc
-flagcxDevNetTerm
-flagcxDevNetWait
-flagcxDevNetWaitCounter
-flagcxDevNetWaitSignal
-flagcxDevNetWaitSignalMeetShadow
+flagcxDevNetSendC
+flagcxDevNetSendS
+flagcxDevNetSignalCtrIncC
+flagcxDevNetSignalCtrIncS
+flagcxDevNetSignalSigAddC
+flagcxDevNetSignalSigAddS
+flagcxDevNetSignalSigIncC
+flagcxDevNetSignalSigIncS
+flagcxDevNetTermC
+flagcxDevNetTermS
+flagcxDevNetWaitC
+flagcxDevNetWaitCounterC
+flagcxDevNetWaitCounterS
+flagcxDevNetWaitS
+flagcxDevNetWaitSignalC
+flagcxDevNetWaitSignalMeetShadowC
+flagcxDevNetWaitSignalMeetShadowS
+flagcxDevNetWaitSignalS
 flagcxGetIntraPointerC
+flagcxGetIntraPointerS
 flagcxGetLocalPointerC
+flagcxGetLocalPointerS
 flagcxGetMulticastPointerC
+flagcxGetMulticastPointerS
 flagcxGetPeerPointerC
-flagcxGetTeamInter
-flagcxGetTeamIntra
-flagcxGetTeamWorld
-flagcxInterBarrierSessionInit
-flagcxInterBarrierSessionSync
-flagcxIntraBarrierSessionArrive
-flagcxIntraBarrierSessionInit
-flagcxIntraBarrierSessionSync
-flagcxIntraBarrierSessionWait
+flagcxGetPeerPointerS
+flagcxGetTeamInterC
+flagcxGetTeamIntraC
+flagcxGetTeamWorldC
+flagcxInterBarrierArriveS
+flagcxInterBarrierSessionInitC
+flagcxInterBarrierSessionSyncC
+flagcxInterBarrierSyncS
+flagcxInterBarrierWaitS
+flagcxIntraBarrierArriveS
+flagcxIntraBarrierSessionArriveC
+flagcxIntraBarrierSessionInitC
+flagcxIntraBarrierSessionSyncC
+flagcxIntraBarrierSessionWaitC
+flagcxIntraBarrierSyncS
+flagcxIntraBarrierWaitS
 flagcxTeamRankToIntraC
+flagcxTeamRankToIntraS
 flagcxTeamRankToWorldC
-flagcxWorldBarrierSessionInit
-flagcxWorldBarrierSessionSync
+flagcxTeamRankToWorldS
+flagcxWorldBarrierArriveS
+flagcxWorldBarrierSessionInitC
+flagcxWorldBarrierSessionSyncC
+flagcxWorldBarrierSyncS
+flagcxWorldBarrierWaitS

bindings/ir/flagcx_device_scalar_ir_impl.h

@@ -19,6 +19,9 @@
 #include "device_utils.h"
 #include "flagcx.h"
 
+// Scalar enum types for IR/Triton integration (shared with native path).
+#include "flagcx_device_enums.h"
+
 // Device traits — provides DeviceAPI with all type/function dispatch.
 #include "comm_traits.h"
 

bindings/ir/flagcx_device_wrapper.h

@@ -0,0 +1,44 @@
+/*************************************************************************
+ * Copyright (c) 2026 BAAI. All rights reserved.
+ *
+ * FlagCX Device API Enums — Scalar enum types for IR/Triton integration.
+ *
+ * These enums encode cooperative group kind and team kind as plain
+ * integers, enabling LLVM IR callers (e.g. Triton) to express these
+ * concepts without instantiating C++ structs.
+ *
+ * Safe to include from: CUDA device code, host code, LLVM bitcode builds.
+ ************************************************************************/
+
+#ifndef FLAGCX_DEVICE_ENUMS_H_
+#define FLAGCX_DEVICE_ENUMS_H_
+
+#include <stdint.h>
+
+/* ================================================================
+ * Cooperative Group Kind
+ *
+ * Identifies the cooperation scope for collective device operations.
+ * Used by the scalar IR API in place of struct-based cooperative groups.
+ * ================================================================ */
+typedef enum {
+  FLAGCX_COOP_THREAD = 0,    /* Single thread (no cooperation) */
+  FLAGCX_COOP_WARP = 1,      /* Full warp (FLAGCX_SIMT_WIDTH threads) */
+  FLAGCX_COOP_BLOCK = 2,     /* Entire CTA */
+  FLAGCX_COOP_TILE_SPAN = 3, /* Consecutive tile span (needs t0, nTiles, id) */
+  FLAGCX_COOP_LANES = 4,     /* Arbitrary lane bitmask */
+} flagcxCoopKind_t;
+
+/* ================================================================
+ * Team Kind
+ *
+ * Identifies the team scope within a communicator.
+ * Used by the scalar IR API in place of struct-based teams.
+ * ================================================================ */
+typedef enum {
+  FLAGCX_TEAM_INTRA = 0, /* Intra-node ranks */
+  FLAGCX_TEAM_INTER = 1, /* Inter-node representatives */
+  FLAGCX_TEAM_WORLD = 2, /* All ranks */
+} flagcxTeamKind_t;
+
+#endif /* FLAGCX_DEVICE_ENUMS_H_ */