diff options
Diffstat (limited to 'src/include/align.h')
-rw-r--r-- | src/include/align.h | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/src/include/align.h b/src/include/align.h new file mode 100644 index 0000000..1c9e7aa --- /dev/null +++ b/src/include/align.h @@ -0,0 +1,19 @@ +/************************************************************************* + * Copyright (c) 2015-2019, NVIDIA CORPORATION. All rights reserved. + * + * See LICENSE.txt for license information + ************************************************************************/ + +#ifndef NCCL_ALIGN_H_ +#define NCCL_ALIGN_H_ + +#define DIVUP(x, y) \ + (((x)+(y)-1)/(y)) + +#define ROUNDUP(x, y) \ + (DIVUP((x), (y))*(y)) + +#define ALIGN_SIZE(size, align) \ + size = ((size + (align) - 1) / (align)) * (align); + +#endif |