saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2
History
Orion Reblitz-Richardson 488ea824ed Additional changes to make GPU builds work (#10507) 
Summary:
A continuation of https://github.com/pytorch/pytorch/pull/10504 for GPU, torch, etc. builds.

I was testing with

```
FULL_CAFFE2=1 python setup.py build_deps | tee ~/log.txt
cat ~/log.txt | egrep 'undefined refer' | sort | less
```

I'll rebase on master when Yangqing's changes in 10504 land, but putting up for some testing.

cc mingzhe09088 anderspapitto ezyang
Pull Request resolved: https://github.com/pytorch/pytorch/pull/10507

Reviewed By: Yangqing

Differential Revision: D9359606

Pulled By: orionr

fbshipit-source-id: c2a3683b3ea5839689f5d2661da0bc9055a54cd2
2018-08-16 13:25:27 -07:00
..
contrib Adding new allreduce bcube routines to ops supported by gloo (#10494) 2018-08-16 10:56:26 -07:00
core Additional changes to make GPU builds work (#10507) 2018-08-16 13:25:27 -07:00
cuda_rtc
db Add db directory for hipifying (#10428) 2018-08-13 10:24:56 -07:00
distributed Additional changes to make GPU builds work (#10507) 2018-08-16 13:25:27 -07:00
experiments Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
ideep Add new MKLDNN fallback operators (#10526) 2018-08-15 15:55:22 -07:00
image Remove caffe1 specific proto (#10380) 2018-08-10 11:10:26 -07:00
mkl s/DISABLE_COPY_AND_ASSIGN/AT_DISABLE_COPY_AND_ASSIGN/ (#10275) 2018-08-07 08:54:26 -07:00
mobile Remove caffe1 specific proto (#10380) 2018-08-10 11:10:26 -07:00
mpi IsType<TensorCPU> -> IsType<Tensor>(CPU) (#10135) 2018-08-03 17:24:59 -07:00
observers IsType<TensorCPU> -> IsType<Tensor>(CPU) (#10135) 2018-08-03 17:24:59 -07:00
onnx
operators Additional changes to make GPU builds work (#10507) 2018-08-16 13:25:27 -07:00
opt Fold AffineChannel to Conv, the same way as BN (for Detectron models) (#10293) 2018-08-13 22:43:37 -07:00
perfkernels Remove caffe1 specific proto (#10380) 2018-08-10 11:10:26 -07:00
predictor build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
proto Additional changes to make GPU builds work (#10507) 2018-08-16 13:25:27 -07:00
python Remove weight from input of onnxifi backend op (#10575) 2018-08-16 10:56:25 -07:00
queue build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
sgd Additional changes to make GPU builds work (#10507) 2018-08-16 13:25:27 -07:00
share
test
transforms
utils Additional changes to make GPU builds work (#10507) 2018-08-16 13:25:27 -07:00
video Remove caffe1 specific proto (#10380) 2018-08-10 11:10:26 -07:00
.clang-format
__init__.py
CMakeLists.txt build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
README.md
release-notes.md
requirements.txt
VERSION_NUMBER

README.md

Caffe2

Jenkins Build Status

Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.

Questions and Feedback

Please use Github issues (https://github.com/caffe2/caffe2/issues) to ask questions, report bugs, and request new features.

Further Resources on Caffe2.ai