strisunshinewentingwang/pytorch

Pull Requests Commits

[tensorexpr] Enabled aten::stack in the fuser pass with static shapes (#74077)

huiguoo committed 2 years ago

90c3699c

[JIT] fix common_expression_hoisting (#74794)

davidberard98 committed 2 years ago

317b1a0e

Add comments for adding shape function and linting (#73570)

Elias Ellison committed 2 years ago

2ef5611f

[skip ci] Set pytree tests to module: pytree owner (#74686)

janeyx99 committed 2 years ago

f9ccf7ab

[skip ci]Revert "Add comments for adding shape function and linting"

malfet committed 2 years ago

3036a030

ci: move nvidia repo disable to common.sh

seemethere committed 2 years ago

b10d7498

Removed python dispatch keys from dispatch key extraction

Chillee committed 2 years ago

3c1dd4e7

Reland "[pytorch][PR] Support dataclasses in TorchScript" take 2 (#74353)

qihqi committed 2 years ago

55477419

[ios] Update _ios-build-test.yml

linbinyu committed 2 years ago

eb3fa2b0

[tensorexpr] Add support for aten::stack (#73801)

huiguoo committed 2 years ago

954c7e2a

ci: Comment out nvidia repositories

seemethere committed 2 years ago

9d9a0f50

[testing] Update dispatch macros (#74977)

malfet committed 2 years ago

Verified bfac65df

Add models test for android and iOS

linbinyu committed 2 years ago

28a4b475

[PyTorchEdge] Make _load_parameters() handle flatbuffer inputs (#74580)

dbort committed 2 years ago

f82b2d4a

[PyTorchEdge] Export flatbuffers from _save_parameters() (#74579)

dbort committed 2 years ago

1659a267

[DataPipe] only apply special serialization when dill is installed

NivekT committed 2 years ago

4c5d5327

Add binary to benchmark model load speed (#74700)

SS-JIA committed 2 years ago

be717775

Revert "Extend CSR constructor to support batched indices and values"

pytorchmergebot committed 2 years ago

cc23725e

Revert "[testing] Update dispatch macros"

pytorchmergebot committed 2 years ago

2e4152b1

[ci] delete unused templates

suo committed 2 years ago

c6102048

[BE] Fix bug in flaky test uploading

janeyx99 committed 2 years ago

b0d7cd01

[Model Averaging] Add a unit test that launches hierarchical SGD by PostLocalSGDOptimizer (#74668)

wayi1 committed 2 years ago

9233af18

Check all CUDA API calls for errors in caffe2/c10/ (#74918)

r-barnes committed 2 years ago

2793cf85

EmbeddingBagCUDA: remove oob check for perf

kshitij12345 committed 2 years ago

273c2f01

[jiterator] kaiser_window

kshitij12345 committed 2 years ago

577eca52

george-qi committed 2 years ago

ef71046f

[testing] Update dispatch macros

khushi-411 committed 2 years ago

eed19a0f

record_function: update to use custom_class API

peterbell10 committed 2 years ago

5630c5ac

[Lint] Clang-format ios folder

malfet committed 2 years ago

3eea311c

[AO][bugfix] Fixing FX QAT but for untraceable modules (#74277)

HDCharles committed 2 years ago

bf091f78

Newer Older