strisunshinewentingwang/pytorch

Pull Requests Commits

Catherine Lee committed 2 years ago

da0ba53a

Catherine Lee committed 2 years ago

6cdbc010

Catherine Lee committed 2 years ago

26aaa2c3

Catherine Lee committed 2 years ago

11165bcd

Catherine Lee committed 2 years ago

b88c35c7

almost everything

Catherine Lee committed 2 years ago

31ce1e55

Remove un-used list. And disable pytest for public binding test. (#96684)

albanD committed 2 years ago

7c525823

Expose API to specify custom context manager for checkpoint (#96783)

soulitzer committed 2 years ago

f3db2a63

Add exceptionhandler to more distributed_c10d APIs (#96770)

H-Huang committed 2 years ago

ac7329b3

[CUDA] Use accumulate type to improve accuracy of grid_sample on half precision inputs [v2] (#96586)

xwang233 committed 2 years ago

1716709d

Output peak memory stats from dynamo torchbench perf CI (#95666)

wconstab committed 2 years ago

54cd4a67

Use .to instead of contiguous to generate channels last tensor (#96791)

eellison committed 2 years ago

44586312

[functorch] batch rule : few decomposition ops (#96744)

kshitij12345 committed 2 years ago

24c49dbf

add to functorch codeowner (#96746)

kshitij12345 committed 2 years ago

9b1b3fdd

[doc] fix `torch.cuda.mem_get_info` doc (#96621)

stas00 committed 2 years ago

11e708dd

Temporarily disable test_ops_jit on Windows (#96859)

huydhn committed 2 years ago

6339ee5d

add 2d tests to ci (#96711)

wanchaol committed 2 years ago

aa09f889

Fixes a layer_norm_nested backwards edge case. (#96788)

drisspg committed 2 years ago

5612aa6a

Fix type hint for `torch.Tensor.grad_fn` (#96804)

XuehaiPan committed 2 years ago

80e8e41c

Fix build, shadowed variable (#96778)

eellison committed 2 years ago

a7d2e451

[aot autograd] avoid cloning some inputs unnecessarily when they dont require grad (#96342)

bdhirsh committed 2 years ago

e9d9151e

[aot autograd] refactor to make functionalization self-contained (#96341)

bdhirsh committed 2 years ago

3cd9c7a1

[Optim in backward] register_hook=False API (#95096)

rohan-varma committed 2 years ago

8e628726

aot autograd: consolidate metadata (#96340)

bdhirsh committed 2 years ago

8ce8d49c

aot_autograd: dont requires_grad on tangents (#96339)

bdhirsh committed 2 years ago

070cefae

aot autograd refactor: make all synthetic base logic layered in a single location (#96235)

bdhirsh committed 2 years ago

a2694699

[aot_autograd] only performance functionalization analysis pass once (#95992)

bdhirsh committed 2 years ago

7a076b7b

Revert "[memory profiling] add a facility to gather combined C++/Python/TorchScript stack traces. (#95541)"

pytorchmergebot committed 2 years ago

e1ea584b

[reland][CI] switch torchbench to a pinned version (#96782)

desertfire committed 2 years ago

33c7be36

COO intersection: allow external hash + hash reuse in sparse_mask (#94596)

nikitaved committed 2 years ago

3fd24fb6

Older