strisunshinewentingwang/pytorch

Pull Requests Commits

[inductor] support SymPy exprs in `reflection_pad2d_backward` lowering (#97604)

nkaretnikov committed 2 years ago

dc45ad70

[inductor] support non-tensor ops with dynamic shapes (#97519)

nkaretnikov committed 2 years ago

9585a7ff

Dynamo stride dim kwargs (#97444)

nima10khodaveisi committed 2 years ago

13dcf635

Add accuracy tests for traced optimizers (#97577)

mrshenli committed 2 years ago

233742cb

Default to aot_eager for torch.compile on MPS (#96980)

msaroufim committed 2 years ago

1b08a013

Enable full train_step tracing and customizable dist graph expansion (#97416)

mrshenli committed 2 years ago

75fb0b6c

Enable lowering to inductor (#96927)

leslie-fang-intel committed 2 years ago

e67b5810

[FSDP] Fix bug in determining whether parameters need to be materialized (#97488)

jamesr66a committed 2 years ago

3b1b585a

[BE] Make `USE_FLASH_ATTENTION` private (#97579)

malfet committed 2 years ago

14177f0d

[vmap] ldl_factor: batch rule (#97518)

kshitij12345 committed 2 years ago

5e014bfb

Rewrite NCCL watchdog to more reliably throw timeout (#97066)

kwen2501 committed 2 years ago

f89af601

Use unordered NEQ comparison for vec512 operator!= implementations (#97466)

bjhargrave committed 2 years ago

ee934fd6

[Better Transformer] make is_causal a hint and force attn_mask to be set on `is_causal=True` in F.MHA (#97214)

Michael Gschwind committed 2 years ago

c757647d

Add myself to nn codeowners (#97277)

mikaylagawarecki committed 2 years ago

2e8086b0

[NCCL] Cleanup NCCL-no-record streams, move to `TORCH_NCCL_AVOID_RECORD_STREAMS` (#97053)

eqy committed 2 years ago

0781188e

[Easy] Apply black to format _spmd files (#97534)

mrshenli committed 2 years ago

021de486

[Easy] Improve error message for meta_mm (#97533)

mrshenli committed 2 years ago

a8f7e0b2

[Kineto] Improve Config Options Part 2 - update to new Kineto Submodule (#97556)

aaronenyeshi committed 2 years ago

b32afbbd

disallow invalid value ranges in torch.testing.make_tensor (#96334)

pmeier committed 2 years ago

129e0390

deprecate low==high in torch.testing.make_tensor (#96333)

pmeier committed 2 years ago

47bfb192

fix low and high in torch.testing.make_tensor for integral inputs (#96124)

pmeier committed 2 years ago

76fb9a1c

only apply domain eps for floating and complex types (#97010)

pmeier committed 2 years ago

779cd1f1

honor low and high for torch.bool in torch.testing.make_tensor (#96332)

pmeier committed 2 years ago

9029361f

fix random mask creation in test_maskedtensor (#97017)

pmeier committed 2 years ago

7602aade

QoL improvements for torch.testing.make_tensor (#96125)

pmeier committed 2 years ago

303eb37e

add proper tests for torch.testing.make_tensor (#96331)

pmeier committed 2 years ago

090af4aa

Revert "Sort: Use cub::WarpMergeSort for small sorts (32 < n <= 128) (#96223)"

pytorchmergebot committed 2 years ago

dbe6da79

fix ignored qualifiers errors (#97443)

mikey dagitses committed 2 years ago

85885301

Inductor: fall back bernoulli on cpu (#97002)

blzheng committed 2 years ago

39c81881

[CI] Add missing --cold-start-latency for the dashboard run (#97547)

desertfire committed 2 years ago

2b75955c

Newer Older