pytorch

88a27ebb - Per Channel Quantization Support for Quantized Linear Operator (#25276)

Per Channel Quantization Support for Quantized Linear Operator (#25276) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/25276 We add the per channel quantization support for the quantized linear operator, based on the recent added per channel quantization APIs in https://github.com/pytorch/pytorch/pull/24935 and https://github.com/pytorch/pytorch/pull/24934. ghstack-source-id: 89267515 Test Plan: buck test mode/dev caffe2/test:quantized -- 'test_qlinear_unpack \(test_quantized\.TestQuantizedLinear\)' --print-passing-details ``` [jianyuhuang@devvm6560.prn2.facebook.com: ~/fbsource/fbcode/caffe2/test] $ buck test mode/dev caffe2/test:quantized -- 'test_qlinear_unpack \(test_quantized\.TestQuantizedLinear\)' --print-passing-details Action graph will be rebuilt because files have been added or removed. Parsing buck files: finished in 1.3 sec Building: finished in 5.7 sec (100%) 8114/8114 jobs, 0 updated Total time: 7.0 sec Trace available for this run at /tmp/testpilot.20190827-141824.842847.log TestPilot test runner for Facebook. See https://fburl.com/testpilot for details. Testpilot build revision c4cde854bae419be71282b0f92bf2d57a9203003 fbpkg f45bf410f1694a6882727cf03961702b at Mon Aug 26 22:10:29 2019 by twsvcscm from /usr/local/fbprojects/packages/testinfra.testpilot/686/t.par Discovering tests Running 1 tests Started new test run: https://our.intern.facebook.com/intern/testinfra/testrun/5629499540372523 ✓ caffe2/test:quantized - test_qlinear_unpack (test_quantized.TestQuantizedLinear) 0.996 1/1 (passed) Test output: > test_qlinear_unpack (test_quantized.TestQuantizedLinear) ... ok > > ---------------------------------------------------------------------- > Ran 1 test in 0.997s > > OK Finished test run: https://our.intern.facebook.com/intern/testinfra/testrun/5629499540372523 Summary (total time 5.05s): PASS: 1 FAIL: 0 SKIP: 0 FATAL: 0 TIMEOUT: 0 OMIT: 0 ``` buck test mode/dev caffe2/test:quantized -- 'test_qlinear \(test_quantized\.TestQuantizedLinear\)' --print-passing-details ``` [jianyuhuang@devvm6560.prn2.facebook.com: ~/fbsource/fbcode/caffe2/test] $ buck test mode/dev caffe2/test:quantized -- 'test_qlinear \(test_quantized\.TestQuantizedLinear\)' --print-passing-details Action graph will be rebuilt because files have been added or removed. Parsing buck files: finished in 0.9 sec Building: finished in 6.4 sec (100%) 8114/8114 jobs, 2 updated Total time: 7.3 sec Trace available for this run at /tmp/testpilot.20190827-141631.836596.log TestPilot test runner for Facebook. See https://fburl.com/testpilot for details. Testpilot build revision c4cde854bae419be71282b0f92bf2d57a9203003 fbpkg f45bf410f1694a6882727cf03961702b at Mon Aug 26 22:10:29 2019 by twsvcscm from /usr/local/fbprojects/packages/testinfra.testpilot/686/t.par Discovering tests Running 1 tests Started new test run: https://our.intern.facebook.com/intern/testinfra/testrun/1125900049005601 ✓ caffe2/test:quantized - test_qlinear (test_quantized.TestQuantizedLinear) 2.893 1/1 (passed) Test output: > test_qlinear (test_quantized.TestQuantizedLinear) ... ok > > ---------------------------------------------------------------------- > Ran 1 test in 2.893s > > OK Finished test run: https://our.intern.facebook.com/intern/testinfra/testrun/1125900049005601 Summary (total time 6.78s): PASS: 1 FAIL: 0 SKIP: 0 FATAL: 0 TIMEOUT: 0 OMIT: 0 ``` buck test mode/dev caffe2/test:quantized -- 'test_qlinear \(test_quantized\.TestDynamicQuantizedLinear\)' --print-passing-details ``` [jianyuhuang@devvm6560.prn2.facebook.com: ~/fbsource/fbcode/caffe2/test] $ buck test mode/dev caffe2/test:quantized -- 'test_qlinear \(test_quantized\.TestDynamicQuantizedLinear\)' --print-passing-details Action graph will be rebuilt because files have been added or removed. Parsing buck files: finished in 1.7 sec Building: finished in 4.9 sec (100%) 8118/8118 jobs, 2 updated Total time: 6.6 sec Trace available for this run at /tmp/testpilot.20190829-153630.613647.log TestPilot test runner for Facebook. See https://fburl.com/testpilot for details. Testpilot build revision f39465ac7f6b26840c8cbd0ae5e367fb8a60ec24 fbpkg cf4e6efcd2fa4642b6f8c26a9bd98d67 at Tue Aug 27 21:58:47 2019 by twsvcscm from /usr/local/fbprojects/packages/testinfra.testpilot/687/t.par Discovering tests Running 1 tests Started new test run: https://our.intern.facebook.com/intern/testinfra/testrun/4222124657066806 ✓ caffe2/test:quantized - test_qlinear (test_quantized.TestDynamicQuantizedLinear) 3.377 1/1 (passed) Test output: > test_qlinear (test_quantized.TestDynamicQuantizedLinear) ... ok > > ---------------------------------------------------------------------- > Ran 1 test in 3.378s > > OK Finished test run: https://our.intern.facebook.com/intern/testinfra/testrun/4222124657066806 Summary (total time 8.18s): PASS: 1 FAIL: 0 SKIP: 0 FATAL: 0 TIMEOUT: 0 OMIT: 0 ``` Differential Revision: D17057818 fbshipit-source-id: 9ad8b9120fd0d9933ca81c132da61b53e2c91b9e