Skip to content

Commit

Permalink
Replace direct include of caffe2.pb.h with an intermediary header caf…
Browse files Browse the repository at this point in the history
…fe2_pb.h (pytorch#10946)

Summary:
Pull Request resolved: pytorch#10946

```
codemod -d . --extensions cc,cpp,cu,cuh,h caffe2/proto/caffe2.pb.h caffe2/proto/caffe2_pb.h
```

Reviewed By: houseroad

Differential Revision: D9539945

fbshipit-source-id: 497d04720e8e7e61c05ffe1b23733d0cb774de7e
  • Loading branch information
ezyang authored and PenghuiCheng committed Sep 11, 2018
1 parent 405c462 commit b9dbe1c
Show file tree
Hide file tree
Showing 98 changed files with 99 additions and 97 deletions.
2 changes: 1 addition & 1 deletion binaries/convert_caffe_image_db.cc
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

#include "caffe2/core/db.h"
#include "caffe2/core/init.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/proto/caffe2_legacy.pb.h"
#include "caffe2/core/logging.h"

Expand Down
2 changes: 1 addition & 1 deletion binaries/convert_db.cc
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

#include "caffe2/core/db.h"
#include "caffe2/core/init.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/core/logging.h"

CAFFE2_DEFINE_string(input_db, "", "The input db.");
Expand Down
2 changes: 1 addition & 1 deletion binaries/convert_encoded_to_raw_leveldb.cc
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@
#include <string>

#include "caffe2/core/init.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/core/logging.h"
#include "leveldb/db.h"
#include "leveldb/write_batch.h"
Expand Down
2 changes: 1 addition & 1 deletion binaries/make_cifar_db.cc
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@
#include "caffe2/core/common.h"
#include "caffe2/core/db.h"
#include "caffe2/core/init.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/core/logging.h"

CAFFE2_DEFINE_string(input_folder, "", "The input folder name.");
Expand Down
2 changes: 1 addition & 1 deletion binaries/make_image_db.cc
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@
#include "caffe2/core/common.h"
#include "caffe2/core/db.h"
#include "caffe2/core/init.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/core/logging.h"

CAFFE2_DEFINE_bool(shuffle, false,
Expand Down
2 changes: 1 addition & 1 deletion binaries/make_mnist_db.cc
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
#include "caffe2/core/common.h"
#include "caffe2/core/db.h"
#include "caffe2/core/init.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/core/logging.h"

CAFFE2_DEFINE_string(image_file, "", "The input image file name.");
Expand Down
2 changes: 1 addition & 1 deletion binaries/print_core_object_sizes_gpu.cc
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
#include "caffe2/core/operator.h"
#include "caffe2/core/context.h"
#include "caffe2/core/context_gpu.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

#define PRINT_SIZE(cls) \
std::cout << "Size of " #cls ": " << sizeof(cls) << " bytes." \
Expand Down
2 changes: 1 addition & 1 deletion binaries/run_plan.cc
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

#include "caffe2/core/init.h"
#include "caffe2/core/operator.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"
#include "caffe2/core/logging.h"

Expand Down
2 changes: 1 addition & 1 deletion binaries/run_plan_mpi.cc
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@

#include "caffe2/core/init.h"
#include "caffe2/core/operator.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"
#include "caffe2/core/logging.h"

Expand Down
2 changes: 1 addition & 1 deletion binaries/speed_benchmark.cc
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
#ifdef CAFFE2_OPTIMIZER
#include "caffe2/opt/optimizer.h"
#endif
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"
#include "caffe2/utils/string_utils.h"

Expand Down
2 changes: 1 addition & 1 deletion binaries/split_db.cc
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@

#include "caffe2/core/db.h"
#include "caffe2/core/init.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/core/logging.h"

CAFFE2_DEFINE_string(input_db, "", "The input db.");
Expand Down
2 changes: 1 addition & 1 deletion binaries/tsv_2_proto.cc
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
#include "caffe2/core/db.h"
#include "caffe2/core/init.h"
#include "caffe2/core/logging.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"

CAFFE2_DEFINE_string(f_in, "", "The input data file name.");
Expand Down
2 changes: 1 addition & 1 deletion caffe2/contrib/tensorrt/tensorrt_tranformer.h
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
#include "caffe2/core/operator.h"
#include "caffe2/core/workspace.h"
#include "caffe2/onnx/onnx_exporter.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "onnx/onnx_pb.h"

namespace caffe2 {
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/blob.h
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
#include "caffe2/core/logging.h"
#include "caffe2/core/tensor.h"
#include "caffe2/core/typeid.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

namespace caffe2 {

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/blob_gpu_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
#include "caffe2/core/blob.h"
#include "caffe2/core/common_gpu.h"
#include "caffe2/core/context_gpu.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include <gtest/gtest.h>

namespace caffe2 {
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/blob_serializer_base.h
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

#include "caffe2/core/common.h"
#include "caffe2/core/registry.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

namespace caffe2 {

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/blob_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
#include "caffe2/core/tensor.h"
#include "caffe2/core/types.h"
#include "caffe2/core/workspace.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"

CAFFE2_DEFINE_int64(caffe2_test_big_tensor_size, 100000000, "");
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/common_cudnn.h
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
#include "caffe2/core/context.h"
#include "caffe2/core/logging.h"
#include "caffe2/core/types.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

#ifndef CAFFE2_USE_CUDNN
#error("This Caffe2 install is not built with cudnn, so you should not include this file.");
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/context.h
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
#include "caffe2/core/event.h"
#include "caffe2/core/logging.h"
#include "caffe2/core/typeid.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

#include "ATen/core/ATenCoreTest.h"
#include "ATen/core/ArrayRef.h"
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/context_base.h
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
#include "caffe2/core/event.h"
#include "caffe2/core/logging.h"
#include "caffe2/core/typeid.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

namespace caffe2 {

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/context_gpu.h
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
#include "caffe2/core/numa.h"
#include "caffe2/core/tensor.h"
#include "caffe2/core/types.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

// Since we are using the macro CAFFE2_USE_CUDNN, we will need to include this
// file after common.h is included.
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/context_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

#include <gtest/gtest.h>
#include "caffe2/core/context.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

namespace caffe2 {

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/db.h
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

#include "caffe2/core/blob_serialization.h"
#include "caffe2/core/registry.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

namespace caffe2 {
namespace db {
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/dispatch/OpSchema.h
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
#pragma once

#include "caffe2/core/dispatch/DispatchKey.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/Array.h"
#include "caffe2/utils/Metaprogramming.h"

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/event.h
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

#include "caffe2/core/common.h"
#include "caffe2/core/logging.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

namespace caffe2 {

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/graph.cc
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
#include "caffe2/core/common.h"
#include "caffe2/core/logging.h"
#include "caffe2/core/net.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

namespace caffe2 {

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/graph.h
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
#pragma once

#include "caffe2/core/common.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"
#include "caffe2/utils/string_utils.h"

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/hip/common_miopen.h
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@
#include "caffe2/core/context.h"
#include "caffe2/core/logging.h"
#include "caffe2/core/types.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

#define MIOPEN_VERSION 1399

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/hip/context_hip.h
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
#include "caffe2/core/numa.h"
#include "caffe2/core/tensor.h"
#include "caffe2/core/types.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

namespace caffe2 {

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/hip/net_async_dag_hip.cc
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
#include "caffe2/core/operator.h"
#include "caffe2/core/static_tracepoint.h"
#include "caffe2/core/timer.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"

#include "caffe2/core/hip/context_hip.h"
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/logging.h
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@

#include <ATen/core/Error.h>
#include "caffe2/core/flags.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

// CAFFE2_LOG_THRESHOLD is a compile time flag that would allow us to turn off
// logging at compile time so no logging message below that level is produced
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/memonger.h
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

#include "caffe2/core/common.h"
#include "caffe2/core/workspace.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

namespace caffe2 {
namespace memonger {
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/net.cc
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
#include "caffe2/core/init.h"
#include "caffe2/core/operator.h"
#include "caffe2/core/timer.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"
#include "caffe2/utils/string_utils.h"

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/net.h
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
#include "caffe2/core/registry.h"
#include "caffe2/core/tensor.h"
#include "caffe2/core/workspace.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/simple_queue.h"
#include "caffe2/utils/thread_pool.h"

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/net_async_base.h
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
#include "caffe2/core/stats.h"
#include "caffe2/core/timer.h"
#include "caffe2/core/workspace.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"
#include "caffe2/utils/thread_pool.h"

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/net_async_dag_gpu.cc
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
#include "caffe2/core/operator.h"
#include "caffe2/core/static_tracepoint.h"
#include "caffe2/core/timer.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"

#include "caffe2/core/context_gpu.h"
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/net_async_dag_gpu.h
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
#include "caffe2/core/common.h"
#include "caffe2/core/net_dag.h"
#include "caffe2/core/workspace.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

namespace caffe2 {

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/net_dag.cc
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
#include "caffe2/core/operator.h"
#include "caffe2/core/static_tracepoint.h"
#include "caffe2/core/timer.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"
#include "caffe2/utils/thread_name.h"

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/net_dag.h
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
#include "caffe2/core/tensor.h"
#include "caffe2/core/timer.h"
#include "caffe2/core/workspace.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/simple_queue.h"

namespace caffe2 {
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/net_dag_utils.cc
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
#include "caffe2/core/operator.h"
#include "caffe2/core/static_tracepoint.h"
#include "caffe2/core/timer.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"

namespace caffe2 {
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/net_dag_utils.h
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
#include "caffe2/core/registry.h"
#include "caffe2/core/tensor.h"
#include "caffe2/core/workspace.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/simple_queue.h"

namespace caffe2 {
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/net_simple.cc
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
#include "caffe2/core/operator.h"
#include "caffe2/core/static_tracepoint.h"
#include "caffe2/core/timer.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"

CAFFE2_DEFINE_bool(
Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/net_simple.h
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
#include "caffe2/core/registry.h"
#include "caffe2/core/tensor.h"
#include "caffe2/core/workspace.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"

namespace caffe2 {

Expand Down
2 changes: 1 addition & 1 deletion caffe2/core/net_simple_async.cc
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
#include "caffe2/core/operator.h"
#include "caffe2/core/static_tracepoint.h"
#include "caffe2/core/timer.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include "caffe2/utils/proto_utils.h"

namespace caffe2 {
Expand Down
Loading

0 comments on commit b9dbe1c

Please sign in to comment.