-
Notifications
You must be signed in to change notification settings - Fork 5.5k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* compile with nccl2 * add ncclGroup; it is necessary in nccl2 * add back libnccl-dev
- Loading branch information
Yang Yang(Tony)
committed
Feb 13, 2018
1 parent
dafc7e3
commit 87f4311
Showing
5 changed files
with
8 additions
and
4 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,4 +1,4 @@ | ||
cc_library(dynamic_loader SRCS dynamic_loader.cc DEPS glog gflags enforce) | ||
nv_library(dynload_cuda SRCS cublas.cc cudnn.cc curand.cc nccl.cc | ||
DEPS dynamic_loader nccl) | ||
DEPS dynamic_loader) | ||
cc_library(dynload_warpctc SRCS warpctc.cc DEPS dynamic_loader warpctc) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters