strisunshinewentingwang/pytorch

Pull Requests Commits

SymIntify cat and narrow (#86191)

wconstab committed 2 years ago

63d8d4f6

Remove softmax from recomputable ops (#86268)

Chillee committed 2 years ago

0e03dc5f

Add refs for torch.unfold and a decomposition for its backward. (#85629)

lezcano committed 2 years ago

c6097688

[FSDP] Dequeue one instead of flush (#86165)

awgu committed 2 years ago

67eb2d59

PixelShuffle check that output is not null before applying kernel (#85155) (#86262)

Thytu committed 2 years ago

1c5ca724

improve annotations (#86105)

pmeier committed 2 years ago

9d6109c4

Memory snapshots from C++ (#86190)

zdevito committed 2 years ago

736adc08

Add opteinsum backend to give users control (#86219)

janeyx99 committed 2 years ago

a348975e

[allocator tracing] missing GIL acquire (#86254)

zdevito committed 2 years ago

db13049b

SymInt fixes from symbolic-shapes branch (#86242)

ezyang committed 2 years ago

d07b8539

[jit][easy] remove deprecated escape sequence (#85987)

davidberard98 committed 2 years ago

ac25c210

Remove `std::cout` from `multinomial_out_mps` (#86246)

malfet committed 2 years ago

2355b625

Remove unnecessary header (#86212)

zou3519 committed 2 years ago

4f95f7ae

enable cpu meta testing (#86226)

albanD committed 2 years ago

6d7235e3

Add ref for cumsum (#86229)

lezcano committed 2 years ago

1432b997

Fix default correction value in std/var decompositions (#85839)

peterbell10 committed 2 years ago

b317736c

[AO] Cubic sparsity level scheduler (#85232)

z-a-f committed 2 years ago

adb12438

[FSDP] Expose internal prefetch limits (#86198)

awgu committed 2 years ago

24879698

Fix ITT unit-tests if PyTorch is compiled with `USE_ITT=OFF` (#86199)

jingxu10 committed 2 years ago

f20e4eab

[6/N] [Dispatchable Collectives] Update recv with CPU / CUDA implementations (#83876)

H-Huang committed 2 years ago

d39e9c1e

[kineto] make ProfilerKineto the only option (#84714)

chaekit committed 2 years ago

d447eff1

Adding Wunused-local-typedef build flag (#86154)

mehtanirav committed 2 years ago

d724a919

[MPS] Remove incorrect asserts from `Copy.mm` (#86184)

malfet committed 2 years ago

8da704cd

Add device logic handling for functions which allow scalar inputs as tensors (#86149)

eellison committed 2 years ago

9da5646c

[primTorch] special: j0, j1, spherical_j0 (#86049)

khushi-411 committed 2 years ago

d6b03085

[easy] Add spaces to vmap over as_strided error message (#86150)

zou3519 committed 2 years ago

8bce2f3d

delete special fake tensor new handling (#86144)

eellison committed 2 years ago

e1859c07

[5/N] [Dispatchable Collectives] Update send with CPU / CUDA implementations (#83859)

H-Huang committed 2 years ago

3f2e7d5c

Move ITT testing to its own test case (#86174)

jingxu10 committed 2 years ago

a75edfa9

Forward fix land race (unexpected successes) (#86186)

Chillee committed 2 years ago

b95e0fcc

Newer Older