mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
* Add missing header "caffe2/core/common.h" before "caffe/proto/caffe.pb.h" to provide CAFFE2_API macro. This only affects the Windows build since CAFFE2_API is only defined for DLL. * Fix ".pb.h" dependency issue about DLL build. CAFFE2_API defined in "caffe2/core/common.h" is required by ".pb.h" generated on Windows for DLL build. We always need to have "#include <caffe2/core/common.h>" before using any proto header. In this case "caffe2.pb.h" is already included by "context_gpu.h" -> "common_cudnn.h" in the correct order, hence we simply remove a line. |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| image_input_op.cc | ||
| image_input_op.h | ||
| image_input_op_gpu.cc | ||
| transform_gpu.cu | ||
| transform_gpu.h | ||