diff --git a/ext-net/example/nccl/net_v3.h b/ext-net/example/nccl/net_v3.h index 26d117c..db1287b 100644 --- a/ext-net/example/nccl/net_v3.h +++ b/ext-net/example/nccl/net_v3.h @@ -5,7 +5,6 @@ #ifndef NCCL_NET_V3_H_ #define NCCL_NET_V3_H_ -#define NCCL_NET_HANDLE_MAXSIZE_V3 64 #define NCCL_NET_MAX_REQUESTS_V3 16 typedef ncclNetProperties_v4_t ncclNetProperties_v3_t; diff --git a/ext-net/example/nccl/net_v4.h b/ext-net/example/nccl/net_v4.h index b39da14..efe4824 100644 --- a/ext-net/example/nccl/net_v4.h +++ b/ext-net/example/nccl/net_v4.h @@ -5,6 +5,8 @@ #ifndef NCCL_NET_V4_H_ #define NCCL_NET_V4_H_ +#define NCCL_NET_HANDLE_MAXSIZE_V4 64 + typedef struct { char* name; // Used mostly for logging. char* pciPath; // Path to the PCI device in /sys.