strisunshinewentingwang/pytorch

Pull Requests Commits

[inductor] Enable an AOTInductor smoketest

desertfire committed 1 year ago

dd383ec8

Recompute flat_arg_fake_tensors after fakification (#98769)

andrewor14 committed 1 year ago

651c1be8

Support >4GB strings in the TorchScript model (#99104)

houseroad committed 1 year ago

df43fef8

[inductor] Refactor post_grad.py (#99127)

jansel committed 1 year ago

6b9a52d1

Add check for same dtype in tensordot implementation (#98938)

kiersten-stokes committed 1 year ago

ae55619a

[quant][fix] Compare resnet with quantizer api with the prepare_fx and decomposed convert flow (#98905)

jerryzh168 committed 1 year ago

9e0df237

Unbreak torch.compile on macos (#99119)

jansel committed 1 year ago

baa06790

Fix MHA doc string (#99146)

felixdivo committed 1 year ago

70072c92

[SPMD] Move some functions to IterGraphModule.setup() (#99076)

fegin committed 1 year ago

286b618b

[SPMD] Remove the unused code (#99075)

fegin committed 1 year ago

d8638765

make ATen/native/cuda/Normalization.cuh data_ptr-correct (#99044)

mikey dagitses committed 1 year ago

9642eb59

make ATen/native/cuda/MultiLabelMarginCriterion.cu data_ptr-correct (#99077)

mikey dagitses committed 1 year ago

46cfde46

trivially convert memcpy sources to use const_data_ptr (#98754)

mikey dagitses committed 1 year ago

4d1297ca

trivially convert all memcpy destinations to mutable_data_ptr (#98753)

mikey dagitses committed 1 year ago

3a510e39

Respect sharded dimensions when aten expaned/view consumes SymInt values (#99058)

mrshenli committed 1 year ago

40aaacd4

[torch package][easy] Make all the save/load tests use buffers (#98798)

kunalb committed 1 year ago

d365d9ed

[MPS] Fix macOS build (#99139)

qqaatw committed 1 year ago

21035462

make ATen/native/cuda/group_norm_kernel.cu data_ptr-correct (#99041)

mikey dagitses committed 1 year ago

05a55b96

Add vmap support for `torch.nn.functional.smooth_l1_loss` (#98357)

yhl48 committed 1 year ago

298cc5c6

Make summarize_perf.py work with perf-compare (#99095)

lezcano committed 1 year ago

1e78a2ed

Don't specialize when indexing by SymInt (#99123)

ezyang committed 1 year ago

ca735ac8

Put GraphArg on the node meta. (#99068)

ezyang committed 1 year ago

0963e118

Expandable blocks in allocator (#96995)

zdevito committed 1 year ago

6a50b83b

[MPS] Add ASSERT_ONLY_METHOD_OPERATORS Finish (#99021)

qqaatw committed 1 year ago

2494e625

[MPS] Add ASSERT_ONLY_METHOD_OPERATORS Part 5 (#99020)

qqaatw committed 1 year ago

7ddeb8d3

[MPS] Add ASSERT_ONLY_METHOD_OPERATORS Part 4 (#99019)

qqaatw committed 1 year ago

70120f2f

[MPS] Add ASSERT_ONLY_METHOD_OPERATORS Part 3 (#99018)

qqaatw committed 1 year ago

f043ff2c

[MPS] Add ASSERT_ONLY_METHOD_OPERATORS Part 2 (#99017)

qqaatw committed 1 year ago

be50c1c4

[MPS] Add ASSERT_ONLY_METHOD_OPERATORS Part 1 (#99016)

qqaatw committed 1 year ago

4ddaab84

make native/cuda/EmbeddingBackwardKernel.cu data_ptr-correct (#99027)

mikey dagitses committed 1 year ago

27049f3f

Older