strisunshinewentingwang/pytorch

Pull Requests Commits

opitimze ConvTransposedND with mkldnn float32 and bfloat16 on CPU (#58348)

mingfeima committed 2 years ago

479e0d64

Additional ops for ShardedTensor, ReplicatedTensor and PartialTensor.

pritamdamania87 committed 2 years ago

9e52b50e

move win cuda tests from pr to trunk

clee2000 committed 2 years ago

c2f362d3

Use scatter_reduce to support masked reductions on sparse COO tensors (sum, prod, amin, amax)

mikaylagawarecki committed 2 years ago

0adf0705

Bugfix NAN and Inf handling for scatter_reduce (amin and amax)

mikaylagawarecki committed 2 years ago

2a9779ad

Allow for custom sharding specs to register their own ops.

pritamdamania87 committed 2 years ago

cc685bcc

add trunc_normal_ function to doc of torch.nn.init

baudzhou committed 2 years ago

fd6991e7

Add support to `Tensor[]?` for structured kernel codegen.

ysiraichi committed 2 years ago

fcf38a58

Change no_torch_function_mode to StashTorchFunctionModeGuard

samdow committed 2 years ago

f23b6291

Add linalg.vander

lezcano committed 2 years ago

621ff0f9

Fix typing for load_state_dict_from_url()

jbschlosser committed 2 years ago

ba818991

tracer compare_outputs should compare in cdouble for complex

zasdfgbnm committed 2 years ago

d20b8397

Add some shape decomps (t, transpose, rot90, stack)

Chillee committed 2 years ago

e9f34931

[SR] Sigmoid out variant calls fast_sigmoid (#75661)

Mike Iovine committed 2 years ago

849984a2

Fix Context.cpp compilation on older Windows

malfet committed 2 years ago

040e2e04

fix package violation caused by D35587412 (#76808)

mikey dagitses committed 2 years ago

37fb636b

Revert "Contribution- Grammatical Corrections in the documentation"

pytorchmergebot committed 2 years ago

8ac6b0a0

Contribution- Grammatical Corrections in the documentation

Sanskar1103 committed 2 years ago

a0ebf1d3

Support torch.Tensor.to for CSR

cpuhrsch committed 2 years ago

ce9a477f

[PyTorch] Make RecordFunction store inputs as ArrayRef (#72484)

swolchok committed 2 years ago

52af4fc5

Introduce distributed checkpoint with ShardedTensor.

Rodrigo Kumpera committed 2 years ago

710246ea

Bugfix scatter_reduce backward formulas

mikaylagawarecki committed 2 years ago

465e0ae2

Load state dict post hook

rohan-varma committed 2 years ago

56bed0dc

Deprecate torch.lu_solve

lezcano committed 2 years ago

f84d4d9c

Deprecate torch.lu

lezcano committed 2 years ago

a5bbfd94

Update torch.lu_unpack docs

lezcano committed 2 years ago

9dc8f256

Add linalg.lu_solve

lezcano committed 2 years ago

fc5b4a5a

[RPC small change] Improving logging for store.wait error

H-Huang committed 2 years ago

31d03c2f

Fix rpc_test.py MyConvNetForMNIST location and sparse tests

pbelevich committed 2 years ago

0a770f80

Handle torch.memory_format serialization in TensorProperties.

Rodrigo Kumpera committed 2 years ago

55e90d17

Newer Older