strisunshinewentingwang/pytorch

Pull Requests Commits

[distributed] implemented find_all_descendants (#102138)

moonbucks committed 2 years ago

06f656c5

[dynamo][higher order op] Support nn.Module calls (#102022)

anijain2305 committed 2 years ago

5d6810a4

[dynamo] Add astunparse dependency (#102120)

desertfire committed 2 years ago

e6af31a5

Segmentation fault in flatbuffers when parsing malformed modules (#95221)

apach301 committed 2 years ago

e6fc7d81

torch.sparse.softmax: allow negative dim (#102172)

nikitaved committed 2 years ago

2e2a7467

Add quantization lowering for nn.PixelShuffle and nn.PixelUnshuffle (#101926)

Riley Dulin committed 2 years ago

424c930f

add ignored_states to FSDP/fully_shard (#102056)

zhaojuanmao committed 2 years ago

956bd038

Revert "Merge type stubs for torch.nn.parallel (#101528)"

pytorchmergebot committed 2 years ago

023bc30b

[spmd] Enable data parallel to work with non 0 batch dim (#100073)

wanchaol committed 2 years ago

d316a2dd

[spmd] add more decomp and fix a sharding bug (#100938)

wanchaol committed 2 years ago

d3788370

[spmd] Improve activation handling, factory ops and batch dim reduction (#100853)

wanchaol committed 2 years ago

dd1f2952

[spmd] enhance batch dim analysis of data parallel (#100852)

wanchaol committed 2 years ago

4d55ea85

[spmd] by default average gradients for nccl backend (#99964)

wanchaol committed 2 years ago

b2eaba6b

[spmd] add option to preserve node types (#100072)

wanchaol committed 2 years ago

942cd12d

No cpp + step current (#102001)

clee2000 committed 2 years ago

2232cce6

Unbind Cat pattern (#101767)

devashishshankar committed 2 years ago

fcf812c3

Merge type stubs for torch.nn.parallel (#101528)

Matthew Hoffman committed 2 years ago

6cabc105

Replace unsqueeze transform with stack (#101766)

devashishshankar committed 2 years ago

fdd28399

Replace split-squeeze pattern (#101765)

devashishshankar committed 2 years ago

c0d0a9f7

[ONNX] Introduce FX-ONNX dispatcher (#100660)

titaiwangms committed 2 years ago

0eb4f072

Refactor normalize passes to use @register_graph_pattern (#101764)

devashishshankar committed 2 years ago

47b41364

[CUDA][DLPack] Handle legacy default streams for DLPack conversion (#101318)

eqy committed 2 years ago

66f6e0e6

[quant][pt2e][be] Move annotate helper function to quantizer/utils.py (#102127)

jerryzh168 committed 2 years ago

3baa67ca

[inductor] Move two cpu tests to test_cpu_repro.py (#101887)

desertfire committed 2 years ago

5f0463a6

[inductor] Added `smooth_l1_loss` refs (#102077)

vfdev-5 committed 2 years ago

e3d97b62

fix inference_mode with torch.compile (#101219)

bdhirsh committed 2 years ago

ddf4f7bc

separate out dynamo .requires_grad and .is_grad_enabled guards (#100570)

bdhirsh committed 2 years ago

98ab11a2

Remove vsx suffix in sleef calls (#100149)

nisanthmp committed 2 years ago

32643bc9

[Reland][functorch] test for compiling functorch transforms (#100718)

zou3519 committed 2 years ago

d08066a4

Add mechanism to turn any RAII guard into a Python Context Manager (#102037)

zou3519 committed 2 years ago

08fb648f

Newer Older