strisunshinewentingwang/pytorch

Pull Requests Commits

Add unsupported prims

davidberard98 committed 2 years ago

64d0117a

Fix ref/decomp filtering

davidberard98 committed 2 years ago

fe0ab8f4

FX Partitioner: use "all aten ops that decompose to prims" as supported NVFuser ops

davidberard98 committed 2 years ago

20d2260e

Added caches for prim_decomp and trace_executor

SherlockNoMad committed 2 years ago

c4737e02

Updated operator support; Add timer

SherlockNoMad committed 2 years ago

d88ae8c8

SherlockNoMad committed 2 years ago

e5ae202e

SherlockNoMad committed 2 years ago

ea06f7c4

[WIP] Interpreter for decomposing aten -> prims

davidberard98 committed 2 years ago

20e254b9

Remove prims from operator_support to avoid confusion

SherlockNoMad committed 2 years ago

59dad716

Address comments

SherlockNoMad committed 2 years ago

17c4bdc5

Cleanup test codes

SherlockNoMad committed 2 years ago

f8047022

Clarify dir structure for fx/passes

SherlockNoMad committed 2 years ago

02308575

Merge branch 'master' into bahuang/graph_partitioner

SherlockNoMad committed 2 years ago

8368e539

optimize `to_dense` for CSC (#79635)

nikitaved committed 2 years ago

9ad91cc6

[CheckpointWrapper] Replace generic mod prefix (#79830)

rohan-varma committed 2 years ago

2ede2872

[JIT] Imbue stringbuf with C locale (#79929)

malfet committed 2 years ago

4b6ba340

refactor partitioner API

SherlockNoMad committed 2 years ago

81fe90d9

Revert "Revert "Add support for multiply on direct SymInt"" (#79681)

Krovatkin committed 2 years ago

fdcb22ef

Revert "Revert "Put symint overloads on a different name"" (#79680)

Krovatkin committed 2 years ago

efc73437

add kineto and flatbuffers to OSS BUCK (#79860)

linbinyu committed 2 years ago

d8b68069

get getitem nodes to same partition as the upstream producer

SherlockNoMad committed 2 years ago

0073f8f8

[xla hash update] update the pinned xla hash (#79587)

pytorchmergebot committed 2 years ago

f1c0280e

fix logical_not reland issues

Chillee committed 2 years ago

e89676f7

error when registering meta kernels to composite ops in core

pytorchmergebot committed 2 years ago

79507d2a

Separate faulty agent RPC tests (#79810)

o-hanna committed 2 years ago

c18a18ca

Fix code that triggers BytesWarning (#79868)

aphedges committed 2 years ago

cb2b7b1e

Add factory function derivatives (#79872)

drisspg committed 2 years ago

a098937c

Fix some links in torch/distributed/CONTRIBUTING.md (#79855)

nariaki3551 committed 2 years ago

6d6e77eb

Minor fix in jit tests to pass TorchDynamo (#79903)

anijain2305 committed 2 years ago

24f550b0

[ao] Adding model report detector base class and implemented detectors

vspenubarthi committed 2 years ago

0656e9e5

Older