strisunshinewentingwang/pytorch

Pull Requests Commits

[dtensor] update README (#89800)

wanchaol committed 1 year ago

12f98f85

update subscriber list (#89799)

wanchaol committed 1 year ago

b09efae3

Add arguments to collect_results (#89611)

williamwen42 committed 1 year ago

f4707ae0

[FSDP] Include module classes in `ModuleWrapPolicy.__repr__` (#89058)

awgu committed 1 year ago

ce17bb95

[FSDP] Limit all gather after pre-unshard (#89057)

awgu committed 1 year ago

c8aaad04

[Checkpoint][2D][5/N] Add checkpoint_utils for distributed checkpoint to testing/_internal/distributed/ (#89873)

wz337 committed 1 year ago

56b3ad78

[FSDP] Remove unneeded stream sync from `clip_grad_norm_()` (#89308)

awgu committed 1 year ago

be80b72a

Generator of tensor inputs with variable layout and structure (batch/non-batch, hybrid/non-hybrid, block/non-block) (#88914)

pearu committed 1 year ago

90bed887

Enable rsqrt (#89771)

eellison committed 1 year ago

275ade63

add env/config flag to disable dynamo (#89828)

mlazos committed 1 year ago

2d32e5dd

[functorch] Move `cond.py` to `_cond.py` and expose `cond()` under functorch.experimental.control_flow. (#89819)

zhxchen17 committed 1 year ago

a70082a8

[FSDP][Easy] Remove outdated TODO (#89217)

awgu committed 1 year ago

d1760d7a

Make fake tensors preserve dense strides in type conversion (#89803)

eellison committed 1 year ago

1a33b7cb

[checkpoint] Improve test (test_nested_dict.py) (#89854)

wz337 committed 1 year ago

9c8a94bf

Fix typo in filesystem.py (#89849)

wz337 committed 1 year ago

cefece37

[dashboaard] Fix flag compilers (#89853)

anijain2305 committed 1 year ago

5a79144a

[CI] Add TorchTrition conda packages (#89841)

malfet committed 1 year ago

59a2fe74

[Caffe2] Fix merge logic bug (#89551)

scramsby committed 1 year ago

24b3b73c

Remove beauby and dzdang from CODEOWNERS (#89811)

kit1980 committed 1 year ago

55789b40

[inductor] Add aten._native_batch_norm_legit to decomposition (#89843)

desertfire committed 1 year ago

693135a9

Update code style for optimizer code (#89862)

mlazos committed 1 year ago

3d47c74c

[quant][docs] Move some of the descriptions out of codeblock (#89795)

jerryzh168 committed 1 year ago

8ca09dda

Enable instance norm running mean test (#89793)

janeyx99 committed 1 year ago

fcb5d6e7

[FSDP] Another fix for `DTensor`, `use_orig_params=True` (#89845)

awgu committed 1 year ago

c599cf24

replace double transpose with single permute in nn.f.mha (#89847)

drisspg committed 1 year ago

b9afa928

Stream actually overrides __new__ so we need to patch it as well (#89592)

albanD committed 1 year ago

8713119c

Move gpu slow tests to sm86 (#87880)

davidberard98 committed 1 year ago

a029ec2c

Deprecating DataPipes (#89794)

ejguan committed 1 year ago

991028cd

Don't unsafely clone autograd meta (#89720)

ezyang committed 1 year ago

6c1fb3f2

Fix CopySlices logic to ensure wrapped node runs properly. (#89812)

albanD committed 1 year ago

02e2eaa9

Newer Older