Commit Graph

15 Commits

Author SHA1 Message Date
Lingyi Liu
ecb05f12c3 Support broadcast for quantized mul kernel (#30442)
Summary:
Since the tensor iterator supports the broadcast, we will just remove the assertion on input shapes.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/30442

Differential Revision: D19976562

Pulled By: lly-zero-one

fbshipit-source-id: 91b27fc8b2570f29d110c6df26eacdd16f587b9f
2020-02-19 16:52:31 -08:00
Zafar Takhirov
a5ac7f6387 Changing observer name
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/27779

Test Plan: Imported from OSS

Differential Revision: D17886605

Pulled By: z-a-f

fbshipit-source-id: 68c50b482e65015336ff27171fd730da493525b6
2019-10-17 11:36:03 -07:00
zou3519
23bffc4f14 Fix most documentation warnings (#27782)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/27782

Warnings show up when running `make html` to build documentation. All of
the warnings are very reasonable and point to bugs in our docs. This PR
attempts to fix most of those warnings.

In the future we will add something to the CI that asserts that there
are no warnings in our docs.

Test Plan: - build and view changes locally

Differential Revision: D17887067

Pulled By: zou3519

fbshipit-source-id: 6bf4d08764759133b20983d6cd7f5d27e5ee3166
2019-10-13 10:34:01 -07:00
Raghuraman Krishnamoorthi
bdcaf6334b Support for add relu functional module (#26612)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/26612

Add support for add relu functional module, this allows for fusion of add and relu quantized operations
ghstack-source-id: 91055976

Test Plan: buck test caffe2/test:quantization -- 'test_functional_module \(test_quantization\.FunctionalModuleTest\)' --print-passing-details

Differential Revision: D17518268

fbshipit-source-id: e1e8b4655d6b32405863ab9d1c7da111fb4343cc
2019-09-30 18:16:58 -07:00
Jerry Zhang
254122dd4e quantize_linear -> quantize_per_tensor (#26574)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/26574

Since we also have `quantized::linear`, `quantize_linear` sounds
confusing, so we plan to rename it before the branch cut

Test Plan:
ci

Imported from OSS

Differential Revision: D17514876

fbshipit-source-id: 01d9005e6ec8cb9950b9d8bba122109c389641d3
2019-09-20 21:58:48 -07:00
Raghuraman Krishnamoorthi
9d06a984f8 Serialization for nn.quantized.functional modules (#25220)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/25220

Add load_from_state_dict and save_from_state_dict for quantized functional modules
ghstack-source-id: 89070836

Test Plan: buck test mode/dev caffe2/test:quantization -- 'test_scriptability_serialization\ \(test_quantization.ScriptabilityTest\)' --print-passing-details

Differential Revision: D17065243

fbshipit-source-id: 413ce0a95d0c27fedb23894f1483e3da2f60f417
2019-08-27 18:56:10 -07:00
Raghuraman Krishnamoorthi
c3c36a5b68 Revert D16923651: Serialization for nn.quantized.functional modules
Test Plan: revert-hammer

Differential Revision:
D16923651

Original commit changeset: eb1234be1941

fbshipit-source-id: c80d0b50db0f949cc293dbc2f825404bbc8cb86c
2019-08-26 15:36:21 -07:00
Zafar Takhirov
3b3261ca8e Adding Scalar add/mul. (#24447)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/24447

Note: This should be landed ONLY after #24259

Pull Request resolved: https://github.com/pytorch/pytorch/pull/24447

Differential Revision: D16846006

Test Plan: Imported from OSS

Pulled By: zafartahirov

fbshipit-source-id: 458fd65279d98cb177ef206240d24dfcbc8d1c1b
2019-08-26 13:05:44 -07:00
Zafar Takhirov
a3e6e82b6c Adding return for the observer in the functional_modules.py
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/25168

Test Plan: Imported from OSS

Differential Revision: D17048164

Pulled By: zafartahirov

fbshipit-source-id: 40ee1f276ee5421255de5b2fc14194402ded10db
2019-08-26 13:03:08 -07:00
Raghuraman Krishnamoorthi
95a3ffc2f1 Serialization for nn.quantized.functional modules (#24924)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/24924

Add load_from_state_dict and save_from_state_dict for quantized functional modules
ghstack-source-id: 89001576

Test Plan: buck test mode/dev caffe2/test:quantization -- 'test_scriptability_serialization\ \(test_quantization.ScriptabilityTest\)' --print-passing-details

Differential Revision: D16923651

fbshipit-source-id: eb1234be1941ccf268a2fc5b756540ab973f3ffb
2019-08-26 12:16:57 -07:00
Zafar Takhirov
9764c2e6f0 Adding quantized mul kernel
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/24444

Test Plan: Imported from OSS

Differential Revision: D16844824

Pulled By: zafartahirov

fbshipit-source-id: 626c40e1cad8329c3d8517156f2d36d7a7472890
2019-08-22 16:54:15 -07:00
Zafar Takhirov
562c5cd73b Adds a placeholder for the 'mul' operator.
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/24421

Test Plan: Imported from OSS

Differential Revision: D16833438

Pulled By: zafartahirov

fbshipit-source-id: 51b20ac060ad657b3f12e4f1cf47369414b342b6
2019-08-16 11:32:51 -07:00
Zafar Takhirov
74ea28322d Replacing axis with dim in quantized cat
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/24151

Test Plan: Imported from OSS

Differential Revision: D16754347

Pulled By: zafartahirov

fbshipit-source-id: c2ebab2f25e0423f16d4f329f98b2e9e138ed549
2019-08-15 12:08:57 -07:00
Jerry Zhang
754bf383b1 Change return type of observer to two tensors (#24339)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/24339

Att

Differential Revision: D16820813

fbshipit-source-id: 3e7301f1700176e19f46e8677a644ba167209254
2019-08-15 10:26:44 -07:00
Zafar Takhirov
4cc16782f3 Removing the make_module script. (#23635)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/23635

It appears it is the same complexity to add new modules using a base class and using a generation script.

Test Plan: Imported from OSS

Differential Revision: D16593364

Pulled By: zafartahirov

fbshipit-source-id: 852dcf41f3dfa2a89152042b8e61d0b6defa8feb
2019-08-13 09:58:28 -07:00