strisunshinewentingwang/pytorch

Pull Requests Commits

Rebase + make more bomb changes

janeyx99 committed 2 years ago

347df614

Make everything pytree friendly

janeyx99 committed 2 years ago

e6a6364f

lint that somehow evaded me before

janeyx99 committed 2 years ago

65b0839c

[BE] Beef up assert_functionalization in test_functionalization.py

janeyx99 committed 2 years ago

025263eb

Enable instance norm running mean test (#89793)

janeyx99 committed 2 years ago

fcb5d6e7

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

awgu committed 2 years ago

c599cf24

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

drisspg committed 2 years ago

b9afa928

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

albanD committed 2 years ago

8713119c

Move gpu slow tests to sm86 (#87880)

davidberard98 committed 2 years ago

a029ec2c

Deprecating DataPipes (#89794)

ejguan committed 2 years ago

991028cd

Don't unsafely clone autograd meta (#89720)

ezyang committed 2 years ago

6c1fb3f2

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

albanD committed 2 years ago

02e2eaa9

[test_nn] split multihead_attention from test_nn (#89748)

kshitij12345 committed 2 years ago

8314d403

[Quant][docs] Use get_default_qconfig_mapping (#87299)

andrewor14 committed 2 years ago

fb47a669

[Quant][fx][bc-breaking] Remove backend_config_utils.py (#89810)

andrewor14 committed 2 years ago

2bce6d09

Revert "[GHA] Decrease Windows test timeout to 120 minutes (#89694)"

pytorchmergebot committed 2 years ago

e1dbd9a2

[FSDP] Relax post-backward assert (#89791)

awgu committed 2 years ago

6e2da426

Revert "Move functorch/_src to torch/_functorch (#88756)"

pytorchmergebot committed 2 years ago

218d9c6e

[follow-up] Python Attr Serialization (#88913)

kshitij12345 committed 2 years ago

086b251f

don't run input mutation analysis in dynamo (#89760)

bdhirsh committed 2 years ago

2f9ec226

[aarch64] add SLEEF dependency for aten_cpu (#89475)

ajaymh committed 2 years ago

3cef87f9

[Quant][docs] Fix BackendConfig example in docstring/README (#89319)

andrewor14 committed 2 years ago

c6ede0bd

Move functorch/_src to torch/_functorch (#88756)

zou3519 committed 2 years ago

52bc5c1c

Guard the boundary of index computed in compute_source_index_and_lambda (#89252)

jgong5 committed 2 years ago

620994cd

[PyTorch Edge] Set training for module only (#89488)

cccclai committed 2 years ago

93772305

Refactoring to share vectorization code for int8/uint8. (#89650)

alexsamardzic committed 2 years ago

a78467f3

[minifier] Continue on assertion for accuracy minification (#89739)

anijain2305 committed 2 years ago

8226a5d3

[dynamo] Don't copy the graph during checkpointing (copy_graphstate) (#89232)

mlazos committed 2 years ago

40dd03ee

add memory_tracker tool to help profiling memory usages (#88825)

zhaojuanmao committed 2 years ago

91899a9e

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

mrshenli committed 2 years ago

7ec7a820

Older