pytorch
908ba05a - [Pytorch] Add python binding to use mobile cpu allocator. (#52376)

Commit
4 years ago
[Pytorch] Add python binding to use mobile cpu allocator. (#52376) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52376 Using default cpu allocator for ops executed on qnnpack backend will result in asan failures with heap overflow since qnnpack (and xnnpack) can access input beyond their and/beginning. Here we are enabling this feature specifically to enable dynamic sparse linear op test using qnnpack engine. In dynamic linear op, the fp32 bias is not packed and hence can result in out-of-bound access. Test Plan: CI Reviewed By: z-a-f Differential Revision: D26491943 fbshipit-source-id: bcc2485e957c7abdef0853c36f6e0f876c20cee3
Author
Parents
Loading