strisunshinewentingwang/pytorch

Pull Requests Commits

anijain2305 committed 1 year ago

7f7089f5

[dynamo] Insert set_rng_state and get_rng_state in the fx graphs

anijain2305 committed 1 year ago

3c3aa34d

[test] Make environment variable name better (#97356)

kiszk committed 1 year ago

f7fe6e14

Add DLPack support for XPU backend by mapping to kDLOneAPI in DLPack … (#94968)

guangyey committed 1 year ago

53c9bc8c

Fix typo under torch/csrc/jit/tensorexpr directory (#97218)

kiszk committed 1 year ago

88234540

[3/n] Consolidate `replicate` and `DDP`: update `replicate` to reuse functions in `DDP` (#96660)

yhcharles committed 1 year ago

721260e9

[BE] Pass `-faligned-new` if supported by compiler (#97887)

malfet committed 1 year ago

af0264ae

fix compiler version detection on MacOS (#97883)

QiangZiBro committed 1 year ago

a95815c6

Fix issue with single input cat (#97822)

jansel committed 1 year ago

1432a893

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

Older