device: change unroll factor

The default value of unroll factor is 2. Changing the unroll
factor to 4 provides better performance for most of the collectives.
This commit is contained in:
Nusrat Islam
2023-05-25 15:42:18 -05:00
والد 58db1cb96d
کامیت 4d1cfb17c8
@@ -11,7 +11,7 @@
#include "collectives.h"
#include "devcomm.h"
#define COLL_UNROLL 2
#define COLL_UNROLL 4
#define NCCL_MAX_DEV_ARITY (NCCL_MAX_TREE_ARITY-1) // Using balanced tree instead of split tree
#define __syncwarp()