strisunshinewentingwang/pytorch

Pull Requests Commits

Sparse CSR: Add backward for torch.sparse.sampled_addmm

IvanYashchuk committed 2 years ago

d7db6a7b

Sparse Compressed tensor factory function 2

pearu committed 2 years ago

e6b4d77c

fix typing in `Module.state_dict` and `load_state_dict`

lkct committed 2 years ago

9fae0762

Add binary_cross_entropy decomp

Chillee committed 2 years ago

9386ea4f

Better error message for android-tests workflow

kit1980 committed 2 years ago

9445609e

[PyTorch] Exercise MHA fast path in JIT

swolchok committed 2 years ago

b182c22e

Add linalg.vander

lezcano committed 2 years ago

1ea49c68

Cleanup ATen-core forward declarations

peterbell10 committed 2 years ago

2e480fc2

Revert "Change 'python mode' to 'torch dispatch mode'"

pytorchmergebot committed 2 years ago

395a620a

Nit in `TripletMarginLoss`

AdityaKane2001 committed 2 years ago

4e4c80d5

adds elementwise opinfos and unary references, extends to out testing

Mike Ruberry committed 2 years ago

c9bd7387

Removed direct doc formatting

vitrioil committed 2 years ago

f92cddd8

Fix lists in the docstring

drbeh committed 2 years ago

76916ccf

Fix failing test when SciPy is not available for test_ldl_factor

IvanYashchuk committed 2 years ago

23dcbe3f

Change 'python mode' to 'torch dispatch mode'

samdow committed 2 years ago

7203a739

[FSDP] Relax exec order valid. to only fwd

awgu committed 2 years ago

5d966e9d

Adds python ref consistency test, elementwise unary reference inputs, and formats test files

Mike Ruberry committed 2 years ago

f6bbecf8

[Nvfuser] Add cast support between double and half types

rdspring1 committed 2 years ago

33be4c94

Change mixed precision API

rohan-varma committed 2 years ago

10bf20da

[complex32] support printing the tensor

kshitij12345 committed 2 years ago

e36d25fb

Port functorch decomps over and fix some tests

Chillee committed 2 years ago

fb246140

Provide an auto wrap policy for common transformer models

zhaojuanmao committed 2 years ago

786903ea

Move functorch decompositions to PyTorch

ezyang committed 2 years ago

a3f10ec2

[fix] mul chalf: cpu scalar and cuda tensor case

kshitij12345 committed 2 years ago

1f1d0b30

Nvfuser faster fallback

jjsjann123 committed 2 years ago

100e72f5

Make Opinfo a dataclass to get a more useful repr

francescocastelli committed 2 years ago

201ddafc

Add better error message for FX when using concrete_args

Chillee committed 2 years ago

2540f866

[CUDA Graphs] Fix OOM inside graph capture_begin

fps7806 committed 2 years ago

b0c5fba9

Revert "[Re-landing 68111] Add JIT graph fuser for oneDNN Graph API (Preview4.1)"

pytorchmergebot committed 2 years ago

3dcd67a1

Allow sharding for distributed tests

janeyx99 committed 2 years ago

0708630d

Newer Older