strisunshinewentingwang/pytorch

Pull Requests Commits

[MPS] Enable fp16 for linear backward (#89774)

kulinseth committed 1 year ago

f2308b1d

[jiterator, complex32] lerp : cuda (#75584)

khushi-411 committed 1 year ago

b5ad9093

beef up inplace/view note on copy slices (#89856)

albanD committed 1 year ago

26054c16

[FSDP][Easy] ufmt `test_fsdp_checkpoint.py` (#89916)

awgu committed 1 year ago

b7c42b40

Fix binary ios builds (#89929)

malfet committed 1 year ago

6e8e7b94

Update Reviewers for PyTorch Distributed team (#89889)

XilunWu committed 1 year ago

1207b0e4

Fix TODOs related to #38095 in test_mps.py (#89815)

kit1980 committed 1 year ago

09f2373e

Revert "Call _sdp_attention in nn.functional.mha (#89470)"

pytorchmergebot committed 1 year ago

f1415b8c

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

pytorchmergebot committed 1 year ago

618a585f

Add a cpp wrapper for Inductor (#88167)

chunyuan-w committed 1 year ago

a6caa9c5

[FSDP][Easy] Remove internal default arg (#89227)

awgu committed 1 year ago

5949d5fe

add bf16 in fp32 out fast path for embedingbag in caffe2 perfkernel (#89198)

zhuhaozhe committed 1 year ago

7cd6e6ac

[benchmarks][dynamo] Trying CI - Set train() for TIMM models accuracy tests (#89780)

anijain2305 committed 1 year ago

68805b08

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

pytorchmergebot committed 1 year ago

969a7d09

Revert "Add bits tensor types (#88594)"

pytorchmergebot committed 1 year ago

4cc5be3a

Row and column select support for block compressed sparse tensors (#88733)

pearu committed 1 year ago

296e1ba4

[PT-D][Checkpoint]Add MultiThreaded FileSystemWriter for distributed checkpointing and Update tests (#87987)

wz337 committed 1 year ago

0cc0e5ef

fix RowwiseMoments vectorization issue on CPU (#84404)

mingfeima committed 1 year ago

87d18cf0

Vectorize erf (#89837)

EikanWang committed 1 year ago

92f08f09

[PT-D][Tensor Parallel] Add more test cases when we use use_orig_params for FSDP wrapping (#89779)

fduwjj committed 1 year ago

009dd3c4

Dynamo, FX, Inductor Progress Bars (#88384)

msaroufim committed 1 year ago

011452a2

[Dynamo] Fix source/reconstruction bugs in NNModule named_* calls (#89729)

yanboliang committed 1 year ago

d88b5555

Update DDP docs for Dynamo/DDPOptimizer (#89096)

wconstab committed 1 year ago

44728375

[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

Newer Older