pytorch

1
load("@rules_proto//proto:defs.bzl", "proto_library")
2
load("@rules_cc//cc:defs.bzl", "cc_library", "cc_proto_library")
3

4
cc_library(
5
    name = "caffe2_pb",
6
    hdrs = ["caffe2_pb.h"],
7
    visibility = [
8
        "//:__pkg__",
9
    ],
10
    deps = [
11
        ":caffe2_cc_proto",
12
        "//c10/core:base",
13
        "//c10/util:base",
14
    ],
15
)
16

17
cc_proto_library(
18
    name = "caffe2_cc_proto",
19
    deps = [":caffe2_proto"],
20
)
21

22
proto_library(
23
    name = "caffe2_proto",
24
    srcs = ["caffe2.proto"],
25
)
26

27
cc_proto_library(
28
    name = "torch_cc_proto",
29
    visibility = ["//:__pkg__"],  # used in torch
30
    deps = [":torch_proto"],
31
)
32

33
proto_library(
34
    name = "torch_proto",
35
    srcs = ["torch.proto"],
36
    deps = [":caffe2_proto"],
37
)
38

39
cc_proto_library(
40
    name = "cc_proto",
41
    visibility = ["//:__pkg__"],
42
    deps = [":proto"],
43
)
44

45
proto_library(
46
    name = "proto",
47
    srcs = [
48
        "caffe2_legacy.proto",
49
        "hsm.proto",
50
        "metanet.proto",
51
        "predictor_consts.proto",
52
        "prof_dag.proto",
53
    ],
54
    deps = [":caffe2_proto"],
55
)
56