pytorch
d200e9de - [Static Runtime] Test for dynamic shapes in SR unit tests (#60579)

Commit
4 years ago
[Static Runtime] Test for dynamic shapes in SR unit tests (#60579) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60579 - Modify testStaticRuntime to take two sets of inputs so if the second set of inputs have bigger shapes, it would trigger memory allocations in resize_ calls. - Modify test scripts so that the output of the test op is managed by the memory planner, as explained in comments. Reviewed By: ajyu Differential Revision: D29221452 fbshipit-source-id: 09f0f7eb384dc8ca67594f1fa76e1e31392ee6ca
Author
Hao Lu
Parents
Loading