FilesExpand file tree

 master

/

FindNCCL.cmake

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

36 lines (32 loc) · 743 Bytes

 master

/

FindNCCL.cmake

Copy path

Top

File metadata and controls

• Code
• Blame

36 lines (32 loc) · 743 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

# Find the NCCL libraries

#

# The following variables are optionally searched for defaults

# NCCL_ROOT_DIR: Base directory where all NCCL components are found

#

# The following are set after configuration is done:

# NCCL_FOUND

# NCCL_INCLUDE_DIR

# NCCL_LIBRARY

find_path(NCCL_INCLUDE_DIR

NAMES nccl.h

PATHS

ENV CUDA_PATH

ENV NCCL_ROOT_DIR

PATH_SUFFIXES

include)

find_library(NCCL_LIBRARY

NAMES nccl

PATHS

ENV CUDA_PATH

ENV NCCL_ROOT_DIR

PATH_SUFFIXES

lib64

lib)

include(FindPackageHandleStandardArgs)

find_package_handle_standard_args(

NCCL

FOUND_VAR NCCL_FOUND

REQUIRED_VARS NCCL_LIBRARY NCCL_INCLUDE_DIR)

mark_as_advanced(

NCCL_INCLUDE_DIR

NCCL_LIBRARY)