Tags: stepsma/pytorch
Tags
Use blas from conda-force See if it helps with pytorch#87148
Store type(module) in the module stack [ghstack-poisoned]
Lift the maximum depth of the Python interpreter stack to adapt large… …/deep models
Update on "[xnnpack][lite-int][graph-build] graph passes and op check… …ing" Beginning of building the xnnpack graph from the torchscript IR. We first massage the torchscript graph using a few graph passes that perform things such as unused self argument removal and constant propagation. This also performs tracing for us so that the model does not have to be prepped by tracing before being lowered by us. The other check we perform is through the torchscript IR to identify any nodes that are not lowerable/supported, and throwing an error to spit out the specific nodes that are not lowerable. Differential Revision: [D39838338](https://our.internmc.facebook.com/intern/diff/D39838338/) **NOTE FOR REVIEWERS**: This PR has internal Meta-specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D39838338/)! [ghstack-poisoned]
Back out "Revert D40198461: [pytorch][PR] Backport currently dont wor… …k with some models if:" Summary: reland after fixing windows build failure for OVR. Notable change: ``` #if defined(FBCODE_CAFFE2) or defined(FB_XPLAT_BUILD) ``` changed to ```#if defined(FBCODE_CAFFE2) || defined(FB_XPLAT_BUILD) ``` Appearently `-DFB_XPLAT_BUILD` wasn't getting picked up in windows if using `or `to connect Original commit changeset: 7a31fc4b455f Original Phabricator Diff: D40198461 Test Plan: waitforsandcastle Reviewed By: davidberard98, cccclai Differential Revision: D40290932 fbshipit-source-id: b62dc1522bd00b737e492dba539e1763247b0504
Update on "functionalization: skip meta reference compute for aot aut… …ograd" [ghstack-poisoned]
[Profiler][trivial] Add profiler options to trace metadata Summary: Add profiler options (`profile_memory`, `record_shapes`, `with_stack`, `with_modules`, and `with_flops`) to trace metadata Test Plan: CI tests Differential Revision: D40373514 fbshipit-source-id: c2fb16a8388c3ad3bc5adf8a1f3ad3f463497ac1
PreviousNext