pytorch/caffe2/python/predictor
Shane Li 620ff25bdb Enhance cpu support on gloo based multi-nodes mode. (#11330)
Summary:
1. Add some gloo communication operators into related fallback list;
2. Work around to avoid compiling errors while using fallback operator whose CPU operator inherits from 'OperatorBase' directly like PrefetchOperator;
3. Add new cpu context support for some python module files and resnet50 training example file.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/11330

Reviewed By: yinghai

Differential Revision: D13624519

Pulled By: wesolwsk

fbshipit-source-id: ce39d57ddb8cd7786db2e873bfe954069d972f4f
2019-01-15 11:47:10 -08:00
..
__init__.py Experimental support for setup.py develop mode install 2018-02-12 23:36:18 -08:00
mobile_exporter_test.py Allow mobile exporter to export string arrays (#10017) 2018-08-01 16:09:58 -07:00
mobile_exporter.py Caffe2: Fix for creating entries of external_input in predic_net (#12979) 2018-11-15 22:33:50 -08:00
predictor_exporter_test.py Rename cuda_gpu_id to device_id in DeviceOption (#12456) 2018-10-09 15:54:04 -07:00
predictor_exporter.py Enhance cpu support on gloo based multi-nodes mode. (#11330) 2019-01-15 11:47:10 -08:00
predictor_py_utils.py Update from facebook (#8384) 2018-06-13 13:10:45 -07:00
predictor_test.py Expose Predictor::run_map to Python 2017-10-18 19:32:56 -07:00
serde.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00