strisunshinewentingwang/pytorch

Pull Requests Commits

[Profiler] Memory profiler part 7: Mark inputs (#87567)

Taylor Robie committed 2 years ago

17fa6bf1

[Profiler] Memory profiler part 6: Mark gradients and temporary intermediates. (#87566)

Taylor Robie committed 2 years ago

64c5c77c

[Profiler] Memory profiler part 5: Data flow graph (#87006)

Taylor Robie committed 2 years ago

5f09a6d5

[Profiler] Memory profiler part 4: Select top level torch ops (#86880)

Taylor Robie committed 2 years ago

c3116dd7

[Inductor] Record cpp kernel in PyTorch Profiler (#89367)

Jiong Gong committed 2 years ago

bb77accb

Don't suppress exceptions from backends (#89656)

ezyang committed 2 years ago

36018a6e

put descriptive kernel names behind config (#89697)

Natalia Gimelshein committed 2 years ago

3e20d023

update docstring for torch.linalg.lstsq (#89383)

jlukehubbard committed 2 years ago

591dfffa

Simplify aot_module_simplified by removing top_args/top_kwargs (#89666)

ezyang committed 2 years ago

c9a0cc86

Don't support kwargs at runtime in aot_module_simplified (#89664)

ezyang committed 2 years ago

6168f22f

Delay verify correctness wrapping to call site. (#89662)

ezyang committed 2 years ago

b04dda42

make inductor correctly propagate nans for maximum and minimum (#89612)

Natalia Gimelshein committed 2 years ago

61a3fe4b

Fix typo in segment_reduction_op_gpu.cu (#89647)

eltociear committed 2 years ago

70c0a300

complex: register c10::complex with py::cast (#89680)

kshitij12345 committed 2 years ago

2c0bd85c

Implement old windows in Python (#87082)

alvgaona committed 2 years ago

abb446af

torchdynamo to torch._dynamo in aot_autograd.py (#89385)

jansel committed 2 years ago

95ea47ef

Remove fake_tensor_propagation (#89646)

ezyang committed 2 years ago

69043247

xfail maml test, instead of running it without fake tensor prop (#89645)

ezyang committed 2 years ago

1aa1014b

[vision hash update] update the pinned vision hash (#89667)

pytorchmergebot committed 2 years ago

a048913e

TorchDynamo: weight prepack for single conv (#89209)

XiaobingSuper committed 2 years ago

3b3ebcd0

TorchDynamo: weight prepack for mkl linear (#89109)

XiaobingSuper committed 2 years ago

0c4f3db7

TorchDynamo: weight prepack for onednn convolution external call (#88988)

XiaobingSuper committed 2 years ago

07151a6b

Revert "Dont clone unmutated args in triton autotuning (#89519)" (#89652)

ezyang committed 2 years ago

0884fdab

Reenable fake_tensor_propagation on test_cudnn_rnn (#89644)

ezyang committed 2 years ago

4a16f8cd

Run optimizer tests with fake tensors (#89643)

ezyang committed 2 years ago

fc7dcb68

Force test_rng_state to run with fake tensor prop (#89641)

ezyang committed 2 years ago

9b13508e

Easy: These tests work with fake_tensor_propagation on (#89640)

ezyang committed 2 years ago

c6be06d9

Support unspecialized integers with dynamic shapes (#89639)

ezyang committed 2 years ago

6fb6eb0a

Cond capture with fake tensors actually works; don't raise in this case (#89638)

ezyang committed 2 years ago

0c96841a

[test_nn] split pruning tests from test_nn (#89590)

kshitij12345 committed 2 years ago

d3c012f4

Newer Older