1010#define FLAGCX_IB_COMMON_H_
1111
1212#include "flagcx_net.h"
13- #include "net.h"
1413#include "ibvcore.h"
14+ #include "net.h"
1515#include <pthread.h>
1616#include <stdint.h>
1717
@@ -57,7 +57,6 @@ struct flagcxIbStats {
5757 int fatalErrorCount ;
5858};
5959
60-
6160struct flagcxIbDev {
6261 pthread_mutex_t lock ;
6362 int device ;
@@ -274,7 +273,6 @@ extern pthread_mutex_t flagcxIbLock;
274273extern int flagcxIbRelaxedOrderingEnabled ;
275274extern pthread_t flagcxIbAsyncThread ;
276275
277-
278276// Parameter functions
279277extern int64_t flagcxParamIbGidIndex (void );
280278extern int64_t flagcxParamIbRoceVersionNum (void );
@@ -295,14 +293,24 @@ extern int64_t flagcxParamIbQpsPerConn(void);
295293extern sa_family_t envIbAddrFamily (void );
296294extern void * envIbAddrRange (sa_family_t af , int * mask );
297295extern sa_family_t getGidAddrFamily (union ibv_gid * gid );
298- extern bool matchGidAddrPrefix (sa_family_t af , void * prefix , int prefixlen , union ibv_gid * gid );
296+ extern bool matchGidAddrPrefix (sa_family_t af , void * prefix , int prefixlen ,
297+ union ibv_gid * gid );
299298extern bool configuredGid (union ibv_gid * gid );
300299extern bool linkLocalGid (union ibv_gid * gid );
301300extern bool validGid (union ibv_gid * gid );
302- extern flagcxResult_t flagcxIbRoceGetVersionNum (const char * deviceName , int portNum , int gidIndex , int * version );
303- extern flagcxResult_t flagcxUpdateGidIndex (struct ibv_context * context , uint8_t portNum , sa_family_t af , void * prefix , int prefixlen , int roceVer , int gidIndexCandidate , int * gidIndex );
304- extern flagcxResult_t flagcxIbGetGidIndex (struct ibv_context * context , uint8_t portNum , int gidTblLen , int * gidIndex );
305- extern flagcxResult_t flagcxIbGetPciPath (char * devName , char * * path , int * realPort );
301+ extern flagcxResult_t flagcxIbRoceGetVersionNum (const char * deviceName ,
302+ int portNum , int gidIndex ,
303+ int * version );
304+ extern flagcxResult_t flagcxUpdateGidIndex (struct ibv_context * context ,
305+ uint8_t portNum , sa_family_t af ,
306+ void * prefix , int prefixlen ,
307+ int roceVer , int gidIndexCandidate ,
308+ int * gidIndex );
309+ extern flagcxResult_t flagcxIbGetGidIndex (struct ibv_context * context ,
310+ uint8_t portNum , int gidTblLen ,
311+ int * gidIndex );
312+ extern flagcxResult_t flagcxIbGetPciPath (char * devName , char * * path ,
313+ int * realPort );
306314extern int flagcxIbWidth (int width );
307315extern int flagcxIbSpeed (int speed );
308316extern int flagcxIbRelaxedOrderingCapable (void );
@@ -319,7 +327,6 @@ extern flagcxResult_t flagcxIbGdrSupport(void);
319327extern flagcxResult_t flagcxIbDmaBufSupport (int dev );
320328extern flagcxResult_t flagcxIbFreeRequest (struct flagcxIbRequest * r );
321329
322-
323330static_assert ((sizeof (struct flagcxIbNetCommBase ) % 32 ) == 0 ,
324331 "flagcxIbNetCommBase size must be 32-byte multiple to ensure "
325332 "fifo is at proper offset" );
@@ -333,9 +340,9 @@ static_assert((offsetof(struct flagcxIbSendComm, wrs) % 32) == 0,
333340 "wrs must be 32-byte aligned" );
334341static_assert ((offsetof(struct flagcxIbRecvComm , remFifo ) % 32 ) == 0 ,
335342 "flagcxIbRecvComm fifo must be 32-byte aligned" );
336- static_assert (sizeof ( struct flagcxIbHandle ) < FLAGCX_NET_HANDLE_MAXSIZE ,
337- " flagcxIbHandle size must be smaller than FLAGCX_NET_HANDLE_MAXSIZE" );
338-
343+ static_assert (
344+ sizeof ( struct flagcxIbHandle ) < FLAGCX_NET_HANDLE_MAXSIZE ,
345+ "flagcxIbHandle size must be smaller than FLAGCX_NET_HANDLE_MAXSIZE" );
339346
340347static_assert (MAX_REQUESTS <= 256 , "request id are encoded in wr_id and we "
341348 "need up to 8 requests ids per completion" );
0 commit comments