1 files changed, 1 insertions, 0 deletions

diff --git a/setup.py b/setup.py
index 9d0c542760..2811dc2992 100644
--- a/setup.py
+++ b/setup.py
@@ -682,6 +682,7 @@ if WITH_CUDA:
         "torch/csrc/cuda/comm.cpp",
         "torch/csrc/cuda/python_comm.cpp",
         "torch/csrc/cuda/expand_utils.cpp",
+        "torch/csrc/cuda/lazy_init.cpp",
         "torch/csrc/cuda/serialization.cpp",
     ]
     main_sources += split_types("torch/csrc/cuda/Tensor.cpp", ninja_global)