pytorch
08b95e3c - [Pytorch, Sparsity] Integrate sparse qnnpack operator in framework (#52377)

Commit
4 years ago
[Pytorch, Sparsity] Integrate sparse qnnpack operator in framework (#52377) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52377 Add QNNPACK specific packed params for sparse linear. Add sparse linear dynamic op with appropriate registration. Add python side LinearDynamic module for sparsity. Add tests to validate sparse linear qnnpack kernels. Note that since these test are mostly run on x86 platform and given that 1x4 sparse kernels are implemented both in sse and arm, LinearDynamic at the moment defaults to 1x4 pattern. Plan is to add another diff that will allow a global override for 8x1 pattern such that prepare/convert flow can work for exporting model for mobile. Test Plan: buck run caffe2/torch/fb/model_optimization:sparsity_test Reviewed By: z-a-f Differential Revision: D26491944 fbshipit-source-id: b98839b4c62664e1fabbb0cbeb2e5c1bd5903b4d
Author
Parents
Loading