diff options
| author | Marat Dukhan <marat@fb.com> | 2018-10-25 22:13:18 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-10-25 22:25:10 -0700 |
| commit | 5e73b828bd4c77ee75f57715a075e86bd1575c29 (patch) | |
| tree | e4cad5ed88d4390ecc28c35b6a5a8b555ecc616d /caffe2/operators/quantized | |
| parent | 4870b1b68f7de34bd5ed59c0aedadfc9444fd194 (diff) | |
| download | pytorch-5e73b828bd4c77ee75f57715a075e86bd1575c29.tar.gz pytorch-5e73b828bd4c77ee75f57715a075e86bd1575c29.tar.bz2 pytorch-5e73b828bd4c77ee75f57715a075e86bd1575c29.zip | |
CMake integration for Int8 ops
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/13145
Differential Revision: D10860849
Pulled By: Maratyszcza
fbshipit-source-id: fdbcc23ff9beaeaedfd561176df6cfe87685c1f5
Diffstat (limited to 'caffe2/operators/quantized')
| -rw-r--r-- | caffe2/operators/quantized/CMakeLists.txt | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/caffe2/operators/quantized/CMakeLists.txt b/caffe2/operators/quantized/CMakeLists.txt new file mode 100644 index 0000000000..b1f51a4aa0 --- /dev/null +++ b/caffe2/operators/quantized/CMakeLists.txt @@ -0,0 +1,31 @@ +# ---[ CPU files only +list(APPEND Caffe2_CPU_SRCS + "${CMAKE_CURRENT_SOURCE_DIR}/init_qnnpack.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_add_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_average_pool_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_channel_shuffle_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_concat_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_conv_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_conv_transpose_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_dequantize_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_fc_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_flatten_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_given_tensor_fill_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_leaky_relu_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_max_pool_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_quantize_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_relu_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_reshape_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_resize_nearest_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_roi_align_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_slice_op.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_softmax_op.cc") + +# ---[ CPU test files +list(APPEND Caffe2_CPU_TEST_SRCS + "${CMAKE_CURRENT_SOURCE_DIR}/int8_test.cc" + "${CMAKE_CURRENT_SOURCE_DIR}/int8_roi_align_op_test.cc") + +# ---[ Send the lists to the parent scope. +set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} PARENT_SCOPE) +set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS} PARENT_SCOPE) |