rohithkrn
0d663cec30
Unify cuda and hip device types in Caffe2 python front end ( #14221 )
...
Summary:
Goal of this PR is to unify cuda and hip device types in caffe2 python front end.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14221
Differential Revision: D13148564
Pulled By: bddppq
fbshipit-source-id: ef9bd2c7d238200165f217097ac5727e686d887b
2018-11-29 14:00:16 -08:00
Junjie Bai
f54ab540af
Rename cuda_gpu_id to device_id in DeviceOption ( #12456 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12456
codemod with 'Yes to all'
codemod -d . --extensions h,cc,cpp,cu,py,proto,pbtxt,pb.txt,config cuda_gpu_id device_id
Overload TextFormat::ParseFromString to do string replace when parsing from protobuf format
Reviewed By: Yangqing
Differential Revision: D10240535
fbshipit-source-id: 5e6992bec961214be8dbe26f16f5794154a22b25
2018-10-09 15:54:04 -07:00
Junjie Bai
ff608a9ff3
Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" ( #12232 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12232
Original commit changeset: fca91fea58b7
This adds proper modifications to the DeviceType <->DeviceOption conversion code added in D10033396
Reviewed By: jerryzh168
Differential Revision: D10132473
fbshipit-source-id: 801ef777e2950982cb47b48051b1471a0a91e64b
2018-10-01 21:54:52 -07:00
Rick Ratmansky
3010dc4208
Revert D10123245: Back out "codemod cuda_gpu_id to device_id"
...
Differential Revision:
D10123245
Original commit changeset: d83da8e00a12
fbshipit-source-id: fca91fea58b7df208edc2e218a1d514f9821ec7b
2018-10-01 12:22:36 -07:00
Yang Liu
7d7d336c45
Back out "codemod cuda_gpu_id to device_id"
...
Summary:
Original commit changeset: f5614a5d2607
D9986213 is causing Multifeed Aggregator a [huge performance different](https://our.intern.facebook.com/intern/ads/analyze_canary/412951953278781781/ ) and is blocking aggregator push since last Friday night: https://fburl.com/feedtools/b6izvwjz
We need to land this revert ASAP to unblock aggregator push.
Reviewed By: orionr
Differential Revision: D10123245
fbshipit-source-id: d83da8e00a1250f5d09811a0a587c127e377aab2
2018-10-01 11:31:14 -07:00
Junjie Bai
3eb5940cf5
codemod cuda_gpu_id to device_id ( #12022 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12022
codemod -d . --extensions h,cc,cpp,cu,py,proto,pbtxt,pb.txt,config cuda_gpu_id device_id
codemod with 'Yes to all'
Reviewed By: orionr
Differential Revision: D9986213
fbshipit-source-id: f5614a5d26078817aee8caf79a494abfd6a95ff1
2018-09-27 20:24:53 -07:00
daquexian
df28f5d06e
[Caffe2] Support non peer access in muji and fix bug when reduced_affix is empty ( #6896 )
...
* [Caffe2] Support non peer access in muji
* [Caffe2] Add test for 4 gpus and 2 groups
* [Caffe2] Add comments
* Fix bug when reduced_affix is empty
* Fix typo and add comments about cpu and amd gpu
2018-06-05 03:14:43 +08:00
Orion Reblitz-Richardson
1d5780d42c
Remove Apache headers from source.
...
* LICENSE file contains details, so removing from individual source files.
2018-03-27 13:10:18 -07:00
Yangqing Jia
8286ce1e3a
Re-license to Apache
...
Summary: Closes https://github.com/caffe2/caffe2/pull/1260
Differential Revision: D5906739
Pulled By: Yangqing
fbshipit-source-id: e482ba9ba60b5337d9165f28f7ec68d4518a0902
2017-09-28 16:22:00 -07:00
Thomas Dudziak
60c78d6160
Fixes range/xrange for Python 3
...
Summary: As title
Differential Revision: D5151894
fbshipit-source-id: 7badce5d3122e8f2526a7170fbdcf0d0b66e2638
2017-06-07 00:04:26 -07:00
Aaron Markham
58f7f2b441
doxygen python block added
...
Summary: Closes https://github.com/caffe2/caffe2/pull/226
Differential Revision: D4793550
Pulled By: JoelMarcey
fbshipit-source-id: cc33e58186304fa8dcac2ee9115dcc271d785b1e
2017-03-29 06:46:16 -07:00
Yangqing Jia
d1e9215184
fbsync
2016-10-07 13:08:53 -07:00
Yangqing Jia
559053d3a8
chunky sync
2016-05-13 14:43:48 -07:00
Yangqing Jia
cf7ca23fc1
make caffe2.python build
2016-03-08 16:48:19 -08:00
Yangqing Jia
9ae880bb6f
move pycaffe2 to caffe2.python
2016-03-08 15:45:30 -08:00
