Fix CMakeLists.txt for Int8 python bindings (pytorch#15047)

TerryTsao · facebook-github-bot · commit c2a754c58b0f · 2018-12-11T10:48:47.000-08:00
Summary: Currently in caffe2, one cannot properly fetch the content of Int8 blobs. Upon digging the source code, it turns out that the relevant source code is not being compiled. Adding the source to CMakeLists.txt fixes this issue. First time ever doing a pull request. Please let me know if there's any rule I should follow. Thanks. Pull Request resolved: pytorch#15047 Differential Revision: D13417583 Pulled By: bddppq fbshipit-source-id: dd39575971a3012635edbf97a045d80e4b62a8eb
diff --git a/caffe2/python/CMakeLists.txt b/caffe2/python/CMakeLists.txt
@@ -4,6 +4,7 @@ set(Caffe2_CPU_PYTHON_SRCS
     "/pybind_state_dlpack.cc"
     "/pybind_state_nomni.cc"
     "/pybind_state_registry.cc"
+    "/pybind_state_int8.cc"
 )
 
 if(CAFFE2_USE_MKLDNN)

Original file line number	Diff line number	Diff line change
`@@ -4,6 +4,7 @@ set(Caffe2_CPU_PYTHON_SRCS`
`4`	`4`	`"/pybind_state_dlpack.cc"`
`5`	`5`	`"/pybind_state_nomni.cc"`
`6`	`6`	`"/pybind_state_registry.cc"`
	`7`	`+ "/pybind_state_int8.cc"`
`7`	`8`	`)`
`8`	`9`
`9`	`10`	`if(CAFFE2_USE_MKLDNN)`