Commits

Branch selector

 main

User selector

mrshenli

Datepicker

All time

Commit History

Commits on Apr 2, 2023

• [Easy] Include SPMD and DTensor files in UFMT checks (pytorch#98148)

  Show description for 0217982
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  0217982

• [Easy] Consolidate string startswith checks (pytorch#98147)

  Show description for 347c67d
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  347c67d

Commits on Apr 1, 2023

• [SPMD] Enable fused Adam in full train step tracing (pytorch#98113)

  Show description for e8d3960
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  e8d3960

• Format DTensor dispatch.py and _meta_registrations.py (pytorch#98114)

  Show description for bccf2ef
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  bccf2ef

Commits on Mar 30, 2023

• Enable adam foreach in full train step tracing (pytorch#97897)

  Show description for 9ec6fdb
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  9ec6fdb

• [SPMD] Support foreach optimizers with functionalization (pytorch#97853)

  Show description for 379fb47
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  379fb47

Commits on Mar 29, 2023

• [Easy] Remove unnecessary graph lint (pytorch#97815)

  Show description for 5949d86
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  5949d86

Commits on Mar 28, 2023

• Allow DTensor to trigger collecives before inplace ops (pytorch#97787)

  Show description for c39f1c1
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  c39f1c1

Commits on Mar 25, 2023

• Add accuracy tests for traced optimizers (pytorch#97577)

  Show description for 233742c
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  233742c

• Enable full train_step tracing and customizable dist graph expansion (pytorch#97416)

  Show description for 75fb0b6
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  75fb0b6

• [Easy] Apply black to format _spmd files (pytorch#97534)

  Show description for 021de48
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  021de48

• [Easy] Improve error message for meta_mm (pytorch#97533)

  Show description for a8f7e0b
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  a8f7e0b

Commits on Jan 18, 2023

• Allow DDP to handle custom dataclass forward outputs (pytorch#92334)

  Show description for 0035340
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  0035340

Commits on Dec 22, 2022

• Restore RNG states for composable reentrant activation checkpointing (pytorch#91265)

  Show description for a055426
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  a055426

Commits on Dec 19, 2022

• Allow FSDP to have ignored modules out of wrapped root (pytorch#91079)

  Show description for e5a48da
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  e5a48da

Commits on Dec 13, 2022

• Fix FSDP checkpoint tests (pytorch#90745)

  Show description for d4dda51
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  d4dda51

Commits on Dec 11, 2022

• [FSDP] Fix _pre_forward type annotation (pytorch#90621)

  Show description for 7ec1cb8
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  7ec1cb8

• [FSDP] Allow MixedPrecision to skip inputs (pytorch#90620)

  Show description for 80542ad
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  80542ad

Commits on Dec 10, 2022

• Add global registry to composable API contract (pytorch#90579)

  Show description for a69cdd9
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  a69cdd9

Commits on Dec 9, 2022

• [FSDP] Allow nested FSDP wrapper to use different mixed precision (pytorch#90523)

  Show description for 0824506
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  0824506

Commits on Dec 5, 2022

• Add sharding strategy to fully_shard (pytorch#90192)

  Show description for 5d6aa99
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  5d6aa99

• Add a repro for fully_shard _unshard error (pytorch#90190)

  Show description for e467088
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  e467088

• Remove deprecated flatten_params_wrapper.py from lintrunner config (pytorch#90154)

  Show description for 199b8b6
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  199b8b6

• Fix fully_shard error when policy is not provided (pytorch#90151)

  Show description for 7a08261
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  7a08261

Commits on Dec 3, 2022

• [Easy] Remove unused parametrization (pytorch#90079)

  Show description for e53a0e3
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  e53a0e3

• Test composable checkpoint wrapping FSDP submodules (pytorch#90078)

  Show description for dd060f3
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  dd060f3

Commits on Dec 2, 2022

• Add integration test for composable fully_shard and checkpoint (pytorch#90041)

  Show description for 74a090a
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  74a090a

Commits on Dec 1, 2022

• Add non-reentrant checkpoint to composable APIs (pytorch#90015)

  Show description for 7bd2844
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  7bd2844

• Add error repro test for FSDP ignored modules with mixed precision (pytorch#89971)

  Show description for 5a36d99
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  5a36d99

Commits on Nov 30, 2022

• [Easy][FSDP] Fix pyre error (pytorch#89930)

  Show description for dfbc4e5
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  dfbc4e5

Commits on Nov 29, 2022

• Test FSDP with submodule non-reentrant checkpointing (pytorch#89781)

  Show description for 7ec7a82
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  7ec7a82

Commits on Nov 23, 2022

• Allow Module forward-pre and forward hooks to take kwargs (pytorch#89389)

  Show description for f5d1857
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  f5d1857

Commits on Nov 21, 2022

• [Easy] Use prepend arg to register forward hooks in quantize.py (pytorch#89391)

  Show description for e0251de
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  e0251de

Commits on Oct 29, 2022

• Add composable activation checkpointing (pytorch#87664)

  Show description for d9b6e41
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  d9b6e41

Commits on Oct 28, 2022

• Fix codeowner errors (pytorch#87954)

  Show description for 89fd451
  

  mrshenli

  authored and

  pytorchmergebot

  committed
  89fd451

Pagination

PreviousNext