pytorch/torch/ao/nn/quantized
HDCharles 428cbd7513 [ao] fixing multihead attention convert size (#110407)
Summary: after converting nn.multihead attention we weren't deleting the
old in_proj_weight and in_proj_bias despite not (really) using them.

Test Plan: python test/test_quantization.py -k
"test_custom_module_multi_head_attention"

Reviewers:

Subscribers:

Tasks:

Tags:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110407
Approved by: https://github.com/jerryzh168
2023-10-03 08:49:12 +00:00
..
dynamic [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00
modules [ao] fixing multihead attention convert size (#110407) 2023-10-03 08:49:12 +00:00
reference [BE] Enable ruff's UP rules and autoformat ao/ (#105430) 2023-07-19 13:44:37 +00:00
__init__.py [ao] fixing public v private for torch.ao.nn.X (#87883) 2022-12-15 03:03:07 +00:00
functional.py AO migration: replace torch internal callsites (#94170) 2023-02-07 02:32:23 +00:00