From 2579c4614501631e17f25b1052dfdf3e53a49d85 Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 24 Oct 2022 10:14:45 -0700 Subject: [PATCH] Fix distributed issue by including distributed files --- setup.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index d8fc6928646b..b618b4571fbd 100644 --- a/setup.py +++ b/setup.py @@ -1089,7 +1089,8 @@ def main(): 'include/torch/csrc/autograd/generated/*.h', 'include/torch/csrc/autograd/utils/*.h', 'include/torch/csrc/cuda/*.h', - 'include/torch/csrc/distributed/c10d/exception.h', + 'include/torch/csrc/distributed/c10d/*.h', + 'include/torch/csrc/distributed/c10d/*.hpp', 'include/torch/csrc/distributed/rpc/*.h', 'include/torch/csrc/jit/*.h', 'include/torch/csrc/jit/backends/*.h',