strisunshinewentingwang/pytorch

Pull Requests Commits

support random for custom device (#97420)

heidongxianhua committed 1 year ago

7fc100a2

Skip test_cpp_wrapper on mac (#97911)

jansel committed 1 year ago

3eecca76

Fix a grep-itself bug when checking for GPU healthcheck (#97929)

huydhn committed 1 year ago

f40b2ed5

[Inductor] Remove fb custom ops dependency (#97907)

yanboliang committed 1 year ago

b23cfe54

improve memory footprint of torch.testing.assert_close (#96131)

pmeier committed 1 year ago

2f6c18d1

Revert "Propogate dynamo shape_env to make_fx (#96437)"

pytorchmergebot committed 1 year ago

8e5c5d20

Add support for pin memory on custom device. (#97621)

donnyyou committed 1 year ago

3460b2b7

add various NT ops needed for testing (#97837)

drisspg committed 1 year ago

f603873c

[kineto] init kineto_activity for each event (#97550)

chaekit committed 1 year ago

2b56da13

[dtensor] remove DeviceMesh typing hack guard type imports (#97889)

wanchaol committed 1 year ago

47ce41e7

[cuDNN][cuDNN V8 API] Fix incorrect use of `emplace` in the benchmark cache (#97838)

eqy committed 1 year ago

aa4ea6e1

Refactor TENSOR_MATCH guards to check dim (for NT support) (#97896)

jbschlosser committed 1 year ago

35be5797

Disable modes in preserve_rng_state (#97738)

jansel committed 1 year ago

04ca3a28

Propogate dynamo shape_env to make_fx (#96437)

tugsbayasgalan committed 1 year ago

3a22916c

Fix typos in torch/fx/_compatibility.py (#97618)

fakeYan committed 1 year ago

7257de6e

Update query version for update_expected.py (#97898)

wconstab committed 1 year ago

2f86c9bc

Stop runner service when its GPU crashes (#97585)

huydhn committed 1 year ago

099b2801

Use the latest NVIDIA driver from setup-nvidia (#97840)

huydhn committed 1 year ago

2806fa44

[dynamo, benchmarks] Add inductor-mode (for max-autotune) and warm start options to dynamo benchmarks (#97719)

williamwen42 committed 1 year ago

b93e1f37

[SPMD] Make compile cache the compilation result and add option to perform transformation (#97836)

fegin committed 1 year ago

942e587d

Fix typo under torch/csrc/jit/runtime directory (#97243)

kiszk committed 1 year ago

d70f9c78

[RFC][inductor][index_put] fallback to aten in torch deterministic mode (#96898)

yuguo68 committed 1 year ago

1f71ac78

[Dynamo] Fix for tuple construction from tuple iterators (#97862)

mlazos committed 1 year ago

e6909f6c

Simplify by using yield from (#97831)

kit1980 committed 1 year ago

477f3f55

Update ufmt to v2.1.0 (#97900)

Skylion007 committed 1 year ago

22b72313

Add config setting to error on recompile (#97829)

mlazos committed 1 year ago

e626be79

Delete fusions_possible counter (#97881)

ezyang committed 1 year ago

bb40b625

[BE][autograd Function] Raise an error if input is returned as-is and saved for forward or backward in setup_context (#97212)

soulitzer committed 1 year ago

313db584

Revert "[dtensor] remove typing hack of DeviceMesh (#94526)"

pytorchmergebot committed 1 year ago

4114c1ea

Refactor dynamic dims api, stateless internals, higher level export API (#96699)

ezyang committed 1 year ago

8372c5dc

Newer Older