From 7d02a1fbc70d324bf587ccbfe508cd2e33ed534a Mon Sep 17 00:00:00 2001 From: Sebastian Messmer Date: Wed, 6 Mar 2019 23:50:14 -0800 Subject: [PATCH] caffe2:libtorch_cuda depends on caffe2:caffe2_gpu (#17729) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17729 When doing "import torch" in fbcode, previously the caffe2 cuda kernels weren't loaded because libcaffe2_gpu.so wasn't loaded. Once you also did "from caffe2.python import workspace", then the cuda kernels were loaded because that triggered a runtime mechanism for loading libcaffe2_gpu.so. We want the cuda kernels to always be available, so this diff adds a dependency from caffe2:libtorch_cuda to caffe2:caffe2_gpu. Reviewed By: ezyang Differential Revision: D14353498 fbshipit-source-id: 76a9fe69f231b308ab40eac393bb216c6fad3658 --- tools/build_variables.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/build_variables.py b/tools/build_variables.py index 1a9b5cc..52acd27 100644 --- a/tools/build_variables.py +++ b/tools/build_variables.py @@ -224,6 +224,7 @@ def add_torch_libs(): "//caffe2/aten:ATen", "//caffe2/aten:generated-aten-headers-cuda", "//caffe2/caffe2:caffe2_cpu", + "//caffe2/caffe2:caffe2_gpu", "//caffe2/torch/lib/libshm:libshm", ], external_deps=[ -- 2.7.4