pytorch/caffe2/python
Michael Antonov a6949abb15 Guard all Caffe2 protobuf string serializations with CAFFE_ENFORCE (fixed reverted bug) (#12848)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12848

Updated all non-test uses of protobuf::MessageLite::SerializeAsString to call
SerializeAsString_EnforceCheck so that the return value is checked and can
throw an exception if failing.

Most of the affected code was called from classes derived from  BlobSerializeBase.
Didn't touch most tests and ENFORCE calls because they usually do checks
anyway.

Original commit changeset: c0760e73ecc7

Reviewed By: dzhulgakov

Differential Revision: D10453456

fbshipit-source-id: d2f2b7b4578e721924354149f08f627c7e3bf070
2018-10-23 16:21:26 -07:00
..
docs adapting caffe2 operator docs generator to pytorch url 2018-10-11 12:55:06 -07:00
examples Add resnext model to OSS (#11468) 2018-09-12 15:59:20 -07:00
helpers
ideep Add IDEEP unit test with zero-dim tensors (#8459) 2018-10-19 23:57:33 -07:00
layers Revert D10123465: Set the correct engine name for position weighted pooling when fp16 is used for training 2018-10-15 01:53:48 -07:00
mint move flags to c10 (#12144) 2018-10-04 02:09:56 -07:00
mkl Re-enable IDEEP graph rewrite test (#12661) 2018-10-15 15:50:28 -07:00
modeling diagnose option: get_entry to print a whole row (#11308) 2018-09-06 21:26:30 -07:00
models Add resnext model to OSS (#11468) 2018-09-12 15:59:20 -07:00
onnx Add attribute exhaustive_search in _blacklist_caffe2_args (#12805) 2018-10-22 22:48:31 -07:00
operator_test caffe2: UpsampleBilinear support for scales (#12736) 2018-10-19 08:55:55 -07:00
predictor Rename cuda_gpu_id to device_id in DeviceOption (#12456) 2018-10-09 15:54:04 -07:00
rnn
serialized_test Use tempfile during serialized test comparison (#12021) 2018-09-25 20:55:45 -07:00
test
trt ONNXIFI transform (#9569) 2018-07-20 15:09:59 -07:00
__init__.py caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
_import_c_extension.py Completely remove build_aten and use_aten (#10469) 2018-08-20 20:26:42 -07:00
allcompare_test.py
attention.py
benchmark_generator.py
binarysize.py
brew.py
brew_test.py Move tanh function to math (#9328) 2018-07-11 13:59:50 -07:00
build.py
cached_reader.py
caffe_translator.py Fix bug in caffe_translator tool (#10056) 2018-10-11 13:13:12 -07:00
caffe_translator_test.py Remove caffe1 specific proto (#10380) 2018-08-10 11:10:26 -07:00
checkpoint.py Create class constant for string literal 'blob_names' 2018-08-24 22:11:43 -07:00
checkpoint_test.py Revert D9566744: [New Checkpoint] Kill the dummy TaskOutput when task.get_step() (#11164) 2018-08-31 22:25:57 -07:00
CMakeLists.txt Remove useless MKL target (#12783) 2018-10-18 14:03:34 -07:00
cnn.py Rename cuda_gpu_id to device_id in DeviceOption (#12456) 2018-10-09 15:54:04 -07:00
compatibility.py migrating deprecated calls without abc module for containers (#11515) 2018-09-13 15:09:22 -07:00
context.py
context_test.py
control.py
control_ops_grad.py
control_ops_util.py
control_test.py
convert.py New serialization format (#12384) 2018-10-16 16:36:58 -07:00
convert_test.py New serialization format (#12384) 2018-10-16 16:36:58 -07:00
convnet_benchmarks.py
convnet_benchmarks_test.py
core.py Migrate DeviceOption.numa_node_id to DeviceOption.device_id 2018-10-19 12:45:48 -07:00
core_gradients_test.py
core_test.py Rename cuda_gpu_id to device_id in DeviceOption (#12456) 2018-10-09 15:54:04 -07:00
crf.py Productionize CRF layer in PyText (#10362) 2018-08-22 00:25:26 -07:00
crf_predict.py Move crf in caffe2 from fb to oss (#12200) 2018-10-01 18:31:41 -07:00
crf_viterbi_test.py Move crf in caffe2 from fb to oss (#12200) 2018-10-01 18:31:41 -07:00
data_parallel_model.py Support GPU version of Spatial Batch Norm (#11711) 2018-10-18 11:22:13 -07:00
data_parallel_model_test.py Support GPU version of Spatial Batch Norm (#11711) 2018-10-18 11:22:13 -07:00
data_workers.py Fixed log message (#10874) 2018-09-05 09:55:52 -07:00
data_workers_test.py
dataio.py Fixing stop condition on composite reader (#9888) 2018-08-20 03:02:20 -07:00
dataio_test.py
dataset.py
db_file_reader.py
db_test.py
device_checker.py
dlpack.h
dyndep.py
embedding_generation_benchmark.py
experiment_util.py
extension_loader.py Completely remove build_aten and use_aten (#10469) 2018-08-20 20:26:42 -07:00
functional.py Caffe2 Functional enforcing inplace output (#10797) 2018-08-23 22:42:47 -07:00
functional_test.py Add support for specifying device_option in Functional (#9619) 2018-07-24 14:41:59 -07:00
fused_8bit_rowwise_conversion_ops_test.py
gradient_check_test.py
gradient_checker.py make the variable declaration closer to usage 2018-10-12 12:07:08 -07:00
gru_cell.py
hsm_util.py
hypothesis_test.py Remove unsafecoalesce op (#12897) 2018-10-22 09:42:26 -07:00
hypothesis_test_util.py Rename cuda_gpu_id to device_id in DeviceOption (#12456) 2018-10-09 15:54:04 -07:00
ideep_test_util.py
layer_model_helper.py parallize the dense part in event models 2018-08-22 22:40:07 -07:00
layer_model_instantiator.py
layer_parameter_sharing_test.py
layer_test_util.py
layers_test.py Allow use substitute ops for LayerNorm (#12177) 2018-10-11 17:36:10 -07:00
lengths_reducer_fused_8bit_rowwise_ops_test.py
lengths_reducer_rowwise_8bit_ops_test.py
lstm_benchmark.py
memonger.py
memonger_test.py
mkl_test_util.py
model_device_test.py
model_helper.py Rename cuda_gpu_id to device_id in DeviceOption (#12456) 2018-10-09 15:54:04 -07:00
model_helper_test.py keep net type info when generating model complete net (#11032) 2018-09-04 21:10:06 -07:00
modifier_context.py
mpi_python.cc
muji.py Rename cuda_gpu_id to device_id in DeviceOption (#12456) 2018-10-09 15:54:04 -07:00
muji_test.py
net_builder.py
net_builder_test.py
net_drawer.py
net_printer.py Rename cuda_gpu_id to device_id in DeviceOption (#12456) 2018-10-09 15:54:04 -07:00
net_printer_test.py
nomnigraph.py Enable mapping from operatordef to converted node for debugging 2018-10-16 21:03:28 -07:00
nomnigraph_test.py expose delete_node (#12840) 2018-10-19 13:30:50 -07:00
normalizer.py Enable alternative LayerNorm impl in FisherGan (#12178) 2018-10-11 17:36:11 -07:00
normalizer_context.py
normalizer_test.py
numa_benchmark.py Migrate DeviceOption.numa_node_id to DeviceOption.device_id 2018-10-19 12:45:48 -07:00
numa_test.py Migrate DeviceOption.numa_node_id to DeviceOption.device_id 2018-10-19 12:45:48 -07:00
observer_test.py
optimizer.py Rename cuda_gpu_id to device_id in DeviceOption (#12456) 2018-10-09 15:54:04 -07:00
optimizer_context.py
optimizer_test.py Add Adadelta optimizer to caffe2 (#9088) 2018-07-24 20:09:21 -07:00
optimizer_test_util.py Implementation of Wngrad optimizer caffe2 python wrapper and unit test on least square regression (#9001) 2018-07-13 18:54:52 -07:00
parallel_workers.py
parallel_workers_test.py
parallelize_bmuf_distributed_test.py
pipeline.py SNNTest with Data Preproc Service (#11707) 2018-09-17 21:25:49 -07:00
pipeline_test.py
predictor_constants.py
pybind_state.cc Guard all Caffe2 protobuf string serializations with CAFFE_ENFORCE (fixed reverted bug) (#12848) 2018-10-23 16:21:26 -07:00
pybind_state.h Remove StaticContext (#12547) 2018-10-10 19:41:03 -07:00
pybind_state_dlpack.cc codemod: caffe::float16 -> at::Half (#11785) 2018-09-20 18:55:19 -07:00
pybind_state_dlpack.h Rename cuda_gpu_id to device_id in DeviceOption (#12456) 2018-10-09 15:54:04 -07:00
pybind_state_gpu.cc Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
pybind_state_hip.cc Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
pybind_state_ideep.cc Guard NumPy usage using USE_NUMPY (#11798) 2018-10-04 12:11:02 -07:00
pybind_state_int8.cc Guard NumPy usage using USE_NUMPY (#11798) 2018-10-04 12:11:02 -07:00
pybind_state_nomni.cc expose delete_node (#12840) 2018-10-19 13:30:50 -07:00
pybind_state_registry.cc Move registry fully to c10 (#12077) 2018-09-27 03:09:54 -07:00
pybind_state_registry.h Move registry fully to c10 (#12077) 2018-09-27 03:09:54 -07:00
python_op_test.py
queue_util.py
record_queue.py
recurrent.py
regularizer.py Add GroupL1Norm regularizer (#9115) 2018-07-06 13:26:09 -07:00
regularizer_context.py
regularizer_test.py Add GroupL1Norm regularizer (#9115) 2018-07-06 13:26:09 -07:00
rnn_cell.py Rewrite http://pytorch.org -> https://pytorch.org throughout project (#12636) 2018-10-15 13:03:27 -07:00
schema.py Add util function from core type to dtype (#10716) 2018-08-21 10:55:19 -07:00
schema_test.py Add util function from core type to dtype (#10716) 2018-08-21 10:55:19 -07:00
scope.py
scope_test.py
session.py
session_test.py
sparse_to_dense_mask_test.py
sparse_to_dense_test.py
task.py Revert D9566744: [New Checkpoint] Kill the dummy TaskOutput when task.get_step() (#11164) 2018-08-31 22:25:57 -07:00
test_util.py Enable CircleCI for Linux jobs (#12389) 2018-10-08 17:09:37 -07:00
text_file_reader.py
timeout_guard.py
toy_regression_test.py
transformations.py Enable Conv fusion optimizations in optimizeForIdeep (#9255) 2018-07-16 21:28:50 -07:00
transformations_test.py nomnigraph - easy - some code cleanup for transformations_test (#12101) 2018-10-01 11:31:08 -07:00
tt_core.py
tt_core_test.py
utils.py migrating deprecated calls without abc module for containers (#11515) 2018-09-13 15:09:22 -07:00
visualize.py
workspace.py New serialization format (#12384) 2018-10-16 16:36:58 -07:00
workspace_test.py Add workspace.RunPlanInBackground (#9637) 2018-07-20 14:56:12 -07:00