[quant][graphmode] Fold quantized prepacking ops#35077
[quant][graphmode] Fold quantized prepacking ops#35077jerryzh168 wants to merge 26 commits intogh/jerryzh168/258/basefrom
Conversation
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: [ghstack-poisoned]
…per_channel pure" Summary: We want to do constant propagation through these ops Test Plan: see next PR: #35077 Reviewers: . Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: We want to do constant propagation through these ops Test Plan: see next PR: #35077 Reviewers: . Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: [ghstack-poisoned]
💊 CircleCI build failures summary and remediationsAs of commit a65ba73 (more details on the Dr. CI page): ✅ None of the build failures appear to be your fault 💚
🚧 1 upstream failure:These were probably caused by upstream breakages:
This comment was automatically generated by Dr. CI (expand for details).Follow this link to opt-out of these comments for your Pull Requests.Please report bugs/suggestions on the GitHub issue tracker. This comment has been revised 110 times. |
| d.dedup(); | ||
| } | ||
|
|
||
| void FoldQuantizedPrepackingOps(Module& module) { |
There was a problem hiding this comment.
I thought coding convention was camelcase for function names.
There was a problem hiding this comment.
I guess I also have mixed that up on some of my passes.
There was a problem hiding this comment.
yeah we are following camelcase for methods and helper functions I think. For jit passes I saw we are using Pascal case, e.g.: https://bddppq.github.io/codebrowser/pytorch/pytorch/torch/csrc/jit/passes/constant_propagation.h.html
I'm not sure why though
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: Differential Revision: [D20655301](https://our.internmc.facebook.com/intern/diff/D20655301) [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: Differential Revision: [D20655301](https://our.internmc.facebook.com/intern/diff/D20655301) [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: Differential Revision: [D20655301](https://our.internmc.facebook.com/intern/diff/D20655301) [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: Differential Revision: [D20655301](https://our.internmc.facebook.com/intern/diff/D20655301) [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: Differential Revision: [D20655301](https://our.internmc.facebook.com/intern/diff/D20655301) [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: Differential Revision: [D20655301](https://our.internmc.facebook.com/intern/diff/D20655301) [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: Differential Revision: [D20655301](https://our.internmc.facebook.com/intern/diff/D20655301) [ghstack-poisoned]
Summary: Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Reviewers: . Subscribers: Tasks: Tags: Differential Revision: [D20655301](https://our.internmc.facebook.com/intern/diff/D20655301) [ghstack-poisoned]
|
This pull request has been merged in 76a8d30. |
Summary: Pull Request resolved: pytorch#35077 Fold the prepack ops: `quantized::linear_prepack` and `quantized::conv2d_prepack` after `freeze` Test Plan: python test/test_jit.py Imported from OSS Differential Revision: D20655301 fbshipit-source-id: fbb4223323f788c88db7b55cfafda46fad106d49
Stack from ghstack:
quantized::add_scalar_reluandquantized::add_scalar_relu_out#35509 [quant][graphmode] Quantization support forquantized::add_scalar_reluandquantized::add_scalar_relu_outaten::conv3d#35347 [quant][graphmode] Quantization support foraten::conv3dquantized::add#35345 [quant][graphmode][fix] Add filter forquantized::addaten::relua general op #35420 [quant][graphmode] Makeaten::relua general opprepare_scripttakes original qconfig_dict #35335 [quant][graphmode]prepare_scripttakes original qconfig_dictquantized::add_scalar#35334 [quant][graphmode] Quantization support forquantized::add_scalarSummary:
Fold the prepack ops:
quantized::linear_prepackandquantized::conv2d_prepackafterfreezeTest Plan:
python test/test_jit.py
Reviewers:
.
Subscribers:
Tasks:
Tags:
Differential Revision: D20655301