strisunshinewentingwang/pytorch

Pull Requests Commits

Enable FSDP ``use_orig_params=True`` mixed precision training when some ranks have no (non-zero sized) parameter shards (#99175)

speediedan committed 1 year ago

157c8690

[dynamo] Support functools.wraps (#98699)

jansel committed 1 year ago

e9be0b0f

[vision hash update] update the pinned vision hash (#99212)

pytorchmergebot committed 1 year ago

b9426ded

Patch failing slow-test logic for inductor-dynamic (#99182)

wconstab committed 1 year ago

3c4622c0

Graph-break on allowed modules if they have hooks (#97184)

wconstab committed 1 year ago

6eab5e88

[ONNX] Support aten.stack for dynamo_export (#99191)

BowenBao committed 1 year ago

55c71cf9

[ONNX] Introduce Input/Ouptut adapter; Switch to 'DynamoExporter' (#98421)

BowenBao committed 1 year ago

606ce5b6

[composable] Enable replicate + trec_shard overall (#98890)

rohan-varma committed 1 year ago

ef11966a

Back out "[core][pruning][be] rename BaseSparsifier to BasePruner (#98747)" (#99171)

drdarshan committed 1 year ago

e45fa1a5

Reintroduce s390x SIMD support (#99057)

AlekseiNikiforovIBM committed 1 year ago

c130b8a7

aot_autograd: more logging on metadata asserts (#99177)

bdhirsh committed 1 year ago

7cb581d4

Remove filter step (#98969)

clee2000 committed 1 year ago

06ad8d6d

[Vulkan] rewrite quantized add, mul, conv2d and conv2d_relu ops (#97468)

manuelcandales committed 1 year ago

cb231915

[PATCH] Back out "Move functional collectives implementation to python. (#98595) (#99168)

Rodrigo Kumpera committed 1 year ago

a910045a

Fix bug in symbolic shape messages (#99169)

voznesenskym committed 1 year ago

20019f7c

Skip sccache initialization on MacOS (#99121)

huydhn committed 1 year ago

70ec347f

[inductor] Use FakeTensorMode() when creating patterns (#99128)

jansel committed 1 year ago

c0d9a026

DDP forward support custom stream accelerated copy. (#98723)

huihoaan committed 1 year ago

bd07f8d2

Enable cadd_sparse for BFloat16 on CPU (#96767)

itaraban committed 1 year ago

a1074ddf

[dynamo] FakeTensor comparison with "is" instead of "==" (#99134)

angelayi committed 1 year ago

2d542d36

Update InternalMatch in subgraph_rewriter after repeated replacements (#99039)

frost-intel committed 1 year ago

b9d69104

Recompute flat_arg_fake_tensors after fakification (#98769)

andrewor14 committed 1 year ago

651c1be8

Support >4GB strings in the TorchScript model (#99104)

houseroad committed 1 year ago

df43fef8

[inductor] Refactor post_grad.py (#99127)

jansel committed 1 year ago

6b9a52d1

Add check for same dtype in tensordot implementation (#98938)

kiersten-stokes committed 1 year ago

ae55619a

[quant][fix] Compare resnet with quantizer api with the prepare_fx and decomposed convert flow (#98905)

jerryzh168 committed 1 year ago

9e0df237

Unbreak torch.compile on macos (#99119)

jansel committed 1 year ago

baa06790

Fix MHA doc string (#99146)

felixdivo committed 1 year ago

70072c92

[SPMD] Move some functions to IterGraphModule.setup() (#99076)

fegin committed 1 year ago

286b618b

[SPMD] Remove the unused code (#99075)

fegin committed 1 year ago

d8638765

Newer Older