caffe2/python/BREW

python_cc_extension(
  name="caffe2_python",
  srcs=["caffe2_python.cc"],
  hdrs=["caffe2_python.h"],
  deps = [
      "//caffe2:all_available_ops",
      "//caffe2:core",
      "//caffe2/db:db",
  ],
)

python_cc_extension(
  name="caffe2_python_gpu",
  srcs=[
    "caffe2_python.cc",
    "caffe2_python_gpu.cc",
  ],
  hdrs=["caffe2_python.h"],
  deps = [
      "//caffe2:all_available_ops",
      "//caffe2:core",
      "//caffe2/db:db",
  ],
)

python_cc_extension(
  name="caffe2_pybind11",
  srcs=[
    "db.cc",
  ],
  deps=[
      "//caffe2:core",
      "//third_party:pybind11",
  ]
)

py_library(
  name="pycaffe2",
  srcs=Glob(["*.py"], excludes=["*_test.py"]),
  deps=[
      ":caffe2_python",
      ":caffe2_python_gpu",
      "//caffe/proto:caffe_proto_py",
      "//caffe2/proto:caffe2_proto_py",
      "//caffe2/python/mint:mint",
  ],
)

for src in Glob(["*_test.py", "*/*_test.py"]):
    py_test(
        name=src[:-3],
        srcs=[src],
        deps=[
            ":pycaffe2",
        ],
    )