frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Split cpu/gpu in caffe2/distributed + some clean up (#20674)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20674

A few targets in caffe2/caffe2/distribute needs to be split too, otherwise won't compile. Also some clean ups and make select_gpu_type to gpu_library_selector

Differential Revision: D15406019

fbshipit-source-id: 6455ab885b248502b48d4c7565597e00fecfd547
This commit is contained in:
Xiaodong Wang
2019-05-21 10:37:54 -07:00
committed by Facebook Github Bot
parent d7cd2d7a8c
commit b5edeca39d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tools/build_variables.py
View File

@ -3,7 +3,7 @@
# not currently relevant so they are combined into one list.
from __future__ import absolute_import, division, print_function, unicode_literals
load("@bazel_skylib//lib:new_sets.bzl", "sets")
load("//caffe2/caffe2/fb:defs_gpu.bzl", "gpu_library_targets")
load("//caffe2/caffe2/fb:defs_gpu.bzl", "gpu_library_selector")
GENERATED_CPP = [
"Functions.cpp",
@ -347,11 +347,11 @@ def add_torch_libs():
)
# TODO: split it into cpp and cuda parts similarly to libtorch
gpu_library_targets(
gpu_library_selector(
name="_C_impl",
deps=[":_C_impl_cuda"],
deps_cpu=[":_C_impl_cpu"],
merge_only=True,
deps_cuda=[":_C_impl_cuda"],
merge_cpu_deps=False,
)
cpp_library(