tree e18a189c7fdbedc17aa39981ef336b020acc076e
parent b039dd15ce78b349be9caa3f0e4aa4d8068719dc
author Jerry Zhang <jerryzh@fb.com> 1614638288 -0800
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> 1614638755 -0800

[quant][graphmode][fx] Add support for fp16 bmm pattern (#52808) (#53021)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53021

Add support for producing fp16 bmm pattern

Test Plan:
python test/test_quantization.py TestQuantizeFxOps.test_bmm

Imported from OSS

Imported from OSS

Reviewed By: vkuzo

Differential Revision: D26725349

fbshipit-source-id: debee718fc33e562aff3f5664757bb52ee85f651