Merge pull request #757 from nusislam/unroll

device: change unroll factor
Этот коммит содержится в:
Nusrat Islam
2023-05-30 14:19:32 -05:00
коммит произвёл GitHub
родитель 2b1efa9e9a 4d1cfb17c8
Коммит 7519ecb476
+1 -1
Просмотреть файл
@@ -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()