projects / platform / upstream / pytorch.git / commitdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: 16f07d7)
NCCL build fix WITH_DISTRIBUTED=1.
authorGregory Chanan <gchanan@fb.com>
Mon, 1 Apr 2019 22:51:44 +0000 (15:51 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Mon, 1 Apr 2019 22:58:54 +0000 (15:58 -0700)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/18691

Reviewed By: ezyang

Differential Revision: D14706205

Pulled By: gchanan

fbshipit-source-id: 802f19bfd7df3703c0dbce03036e2f2e32eb3efb

torch/csrc/distributed/c10d/ddp.cpp patch | blob | history

diff --git a/torch/csrc/distributed/c10d/ddp.cpp b/torch/csrc/distributed/c10d/ddp.cpp
index c60d077..be1ee78 100644 (file)
--- a/torch/csrc/distributed/c10d/ddp.cpp
+++ b/torch/csrc/distributed/c10d/ddp.cpp
@@ -13,6 +13,7 @@
 #include <ATen/cuda/CUDAEvent.h>
 #include <ATen/cuda/CUDAMultiStreamGuard.h>
 #include <c10/cuda/CUDAGuard.h>
+#include <c10/cuda/CUDACachingAllocator.h>
 
 #include <cstddef>
 #include <memory>
Domain: Machine Learning / ML Framework;