strisunshinewentingwang/pytorch

Pull Requests Commits

Update torchrun and TorchElastic to take optional `local_addr` param to allow skip local IP lookup if specified (#88922)

cz-37 committed 1 year ago

5d37890b

Restructure ShapeEnv so it uses GuardBuilder.SHAPE_ENV directly (#91055)

ezyang committed 1 year ago

57390116

[torchgen] Add CI job to make sure torchgen works for Executorch op registration (#89596)

larryliu0820 committed 1 year ago

2f154f68

[QNNPACK] Add more unaligned attributes (#91208)

digantdesai committed 1 year ago

37ea99cd

[MPS] Fix data type issues in Unary ops (#91120)

razarmehr committed 1 year ago

a274b5b9

[BE] Use `aten` global in `torch._refs` (#91189)

malfet committed 1 year ago

c8546c93

[BE] Use `aten` global var (#91188)

malfet committed 1 year ago

46f64117

[MPS] Fix `torch.std`/`torch.var` default/correction handling (#91203)

malfet committed 1 year ago

dd735b96

Decompose fill, zero, and zeros_like (#90968)

peterbell10 committed 1 year ago

e670c261

Temporarily disable ROCm tests (#91217)

kit1980 committed 1 year ago

eeacb6ae

[generate_vmap_rule] Add generate_vmap_rule to autograd.Function (#90966)

zou3519 committed 1 year ago

2f37804c

[generate_vmap_rule] reductify_leaf helper function (#90965)

zou3519 committed 1 year ago

2a559841

[generate_vmap_rule] Add mechanism to override ctx.saved_tensors (CtxWithSavedTensors) (#90964)

zou3519 committed 1 year ago

53c94ef1

[generate_vmap_rule] add restore_vmap helper function (#90963)

zou3519 committed 1 year ago

31981d01

Revert "[inductor] Rewrite Triton templates + epilogue fusion (retry) (#91105)"

pytorchmergebot committed 1 year ago

94262efc

Delete dead intermediary_symbols (#91070)

ezyang committed 1 year ago

e932c3e5

[Nested Tensor] do not use at::cuda::getDefaultCUDAStream(), again (#91180)

jeffdaily committed 1 year ago

e5a748fe

Minor typing improvements (#91068)

ezyang committed 1 year ago

1c46a32b

Doc improvement in LKJCholesky distribution (#91091)

felixdivo committed 1 year ago

7fecba7b

Update __init__.py (#91196)

richardachen committed 1 year ago

dafd0432

[threaded pg] adapt test_pointwise_ops.py (#90713)

XilunWu committed 1 year ago

712170e9

[threaded pg] make exception handling consistent with MultiProcessTestCase (#90712)

XilunWu committed 1 year ago

a6dcebf9

[threaded pg] add assertion util to MultiThreadedTestCase (#90595)

XilunWu committed 1 year ago

34da4460

[NamedOptimizer][2/N] Prepare the enablement of state_dict for FSDP (#91147)

fduwjj committed 1 year ago

c7e7ea92

[ROCm] Modify GPUs visibility code when starting docker container (#91031)

jithunnair-amd committed 1 year ago

c248f2f3

Update optim.rst (#91195)

richardachen committed 1 year ago

f460893c

use libdevice for tanh (#90889)

ngimel committed 1 year ago

c43209db

refactor the dfs cyclic search from recursion to iterative approach (#91042)

jjsjann123 committed 1 year ago

192a11d4

Remove breakpoint (#91128)

lezcano committed 1 year ago

e6fcf7ad

Small operatorbench changes (#91027)

eellison committed 1 year ago

cdbca356

Newer Older