Skip to content

Integrate FC fp16 exporter into Dper2#26582

Closed
wx1988 wants to merge 1 commit intopytorch:masterfrom
wx1988:export-D17439720
Closed

Integrate FC fp16 exporter into Dper2#26582
wx1988 wants to merge 1 commit intopytorch:masterfrom
wx1988:export-D17439720

Conversation

@wx1988
Copy link

@wx1988 wx1988 commented Sep 20, 2019

Summary:
Enable FC FP16 exporter for predict/eval net.

Add the blob quantization.
replace the op in the eval/predictor net.

Add option to control the use_attribution in the advanced modeling.

Note that:

  • FCTransposed is not quantized as I didn't find this operator used in the existing codebase or model.

Test Plan:

Unit test:

./buck-out/gen/caffe2/caffe2/fb/dper/layer_models/tests/exporter_test#binary.par

./buck-out/gen/fblearner/flow/projects/dper/tests/validators/test_exporter_options_validators#binary.par

Differential Revision: D17439720

@pytorchbot pytorchbot added caffe2 module: pybind Related to our Python bindings / interactions with other Python libraries labels Sep 20, 2019
@wx1988 wx1988 force-pushed the export-D17439720 branch 2 times, most recently from eea0e3a to 6843158 Compare September 29, 2019 01:27
Summary:
Pull Request resolved: pytorch#26582

Add the blob quantization.
replace the op in the eval/predictor net.

Test Plan:
# Unit test:

-----

buck build fblearner/flow/projects/dper/tests/validators:test_exporter_options_validators

./buck-out/gen/fblearner/flow/projects/dper/tests/validators/test_exporter_options_validators#binary.par

----

buck build caffe2/caffe2/fb/dper/layer_models/tests:exporter_test

./buck-out/gen/caffe2/caffe2/fb/dper/layer_models/tests/exporter_test-2.7#binary.par

Reviewed By: chocjy

Differential Revision: D17439720

fbshipit-source-id: 4cec0fcac01073f7df878f8c9416d525014a1466
@facebook-github-bot
Copy link
Contributor

This pull request has been merged in a1513dc.

pdlive215 pushed a commit to pdlive215/pytorch that referenced this pull request Nov 27, 2019
Summary:
Pull Request resolved: pytorch#26582

Add the blob quantization.
replace the op in the eval/predictor net.

Test Plan:
# Unit test:

-----

buck build fblearner/flow/projects/dper/tests/validators:test_exporter_options_validators

./buck-out/gen/fblearner/flow/projects/dper/tests/validators/test_exporter_options_validators#binary.par

----

buck build caffe2/caffe2/fb/dper/layer_models/tests:exporter_test

./buck-out/gen/caffe2/caffe2/fb/dper/layer_models/tests/exporter_test-2.7#binary.par

Reviewed By: chocjy

Differential Revision: D17439720

fbshipit-source-id: 68de5d0322b0111aeca5ed552210bf80a4cddc78
thiagocrepaldi pushed a commit to thiagocrepaldi/pytorch that referenced this pull request Feb 4, 2020
Summary:
Pull Request resolved: pytorch#26582

Add the blob quantization.
replace the op in the eval/predictor net.

Test Plan:
# Unit test:

-----

buck build fblearner/flow/projects/dper/tests/validators:test_exporter_options_validators

./buck-out/gen/fblearner/flow/projects/dper/tests/validators/test_exporter_options_validators#binary.par

----

buck build caffe2/caffe2/fb/dper/layer_models/tests:exporter_test

./buck-out/gen/caffe2/caffe2/fb/dper/layer_models/tests/exporter_test-2.7#binary.par

Reviewed By: chocjy

Differential Revision: D17439720

fbshipit-source-id: 68de5d0322b0111aeca5ed552210bf80a4cddc78
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

caffe2 Merged module: pybind Related to our Python bindings / interactions with other Python libraries

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants