mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
79c5275d75
(1) build file changes. (2) removed data/ subfolder - anything involving datasets should probably be tested separately. (3) Some new functionalities. TODOs: (1) build files for contrib/ (2) cudnn 5.05 compatibility (currently supporting 5.04)
46 lines
880 B
Text
46 lines
880 B
Text
python_cc_extension(
|
|
name="caffe2_python",
|
|
srcs=["caffe2_python.cc"],
|
|
hdrs=["caffe2_python.h"],
|
|
deps = [
|
|
"//caffe2:all_available_ops",
|
|
"//caffe2/core: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:core",
|
|
"//caffe2/db:db",
|
|
],
|
|
)
|
|
|
|
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",
|
|
],
|
|
)
|
|
|