OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
6,162 Commits 8,304 Branches 150 Tags 1.5 GiB
93f49667d0
Commit Graph

4 Commits

Author SHA1 Message Date
Sam Gross
93f49667d0
Allow Variables in calls to NCCL bindings. (#4725) 
The Tensor and Variable classes are being merged in Python. This means
that all interfaces to C++ must accept Variables where they previously
accepted Tensors.
 2018-01-18 15:25:41 -05:00
Sam Gross
23fc2b7e06
Define CHECK in torch/csrc/cuda/nccl.h (#4721) 
The CHECK function was used but not defined in the nccl.h header file.
 2018-01-18 13:08:06 -05:00
Adam Paszke
1061d7970d Move broadcast and broadcast_coalesced to C++ 2018-01-18 11:16:45 +01:00
Adam Paszke
de5f7b725e Base for pure C++ NCCL interface 2018-01-18 11:16:45 +01:00