Michael Lazos

Disallow torch.jit in graph, so that it is properly handled by skipfiles

1

Fixes https://github.com/pytorch/pytorch/issues/90774 by graph breaking on torch.jit* properly I attempted to add support for torch.jit.isinstance, but it ended up requiring allowing all of torch.jit tracing which seems non-productive, graph breaking...

Don't copy name_to_input in OutputGraph

1

This copy isn't necessary and regressed tracing Adam by ~10s with a 1000 parameter model. cc @soumith @voznesenskym @yanboliang @penguinwu @anijain2305 @EikanWang @jgong5 @Guobing-Chen @chunyuan-w @XiaobingSuper @zhuhaozhe @blzheng @Xia-Weiwen @wenzhe-nrv...

component-level configurable logging for dynamo, inductor, aot

32

Summary: Adds NNC-like logging that is configured through an env var `TORCH_COMPILE_LOGS` Examples: `TORCH_LOGS="dynamo,guards" python script.py` - prints dynamo logs at level INFO with guards of all functions that are...

Fix user warning for tensor LR

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #125383 * #123753 * __->__ #123752 * #123751

Update LRScheduler to handle tensor LR

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #125383 * __->__ #123753 * #123752 * #123751 cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78...

Swap warning counter to flag in LRScheduler

5

This was a counter previously, this should be a flag to indicate whether or not the optimizer step has been called. Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #125383 *...

Enable dynamo-traced optimizer peak memory tests

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #124544 * __->__ #124543 * #124542 * #124541 * #124540

LRScheduler composability tests

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #125383 * #123753 * #123752 * #123751 cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78...

Enable dynamo traced test_param_group_with_lrscheduler_goes_right_direction

9

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #124544 * #125826 * #125825

Tighten fallback conditions for compiled optim

1

Since we now will support `capturable=False` when it's valid, narrow the eager fallback conditions to the cases where `compile` will fail. The lone case here is when the user deletes...