strisunshinewentingwang/pytorch

Pull Requests Commits

[sparsity] Lambda Scheduler (#59771)

z-a-f committed 3 years ago

05c1e5b6

[sparsity] Base sparsity level scheduler class (#59770)

z-a-f committed 3 years ago

37ebf2e3

Fix some more loops (#60895)

r-barnes committed 3 years ago

ed63fb52

[DDP] Make uneven inputs work with comm. hook (#61020)

rohan-varma committed 3 years ago

43fb39c3

[DDP] Refactor uneven inputs to take GradBucket (#61019)

rohan-varma committed 3 years ago

94b73068

CTCLoss: Remove dispatching in parallel region (#60599)

peterbell10 committed 3 years ago

512448a4

[sparsity] WeightNormSparsifier (#58955)

z-a-f committed 3 years ago

d42f1751

[sparsity][refactor] Import factoring out (#58707)

z-a-f committed 3 years ago

7ab27294

[sparsity] Sparsifier class (#58704)

z-a-f committed 3 years ago

973e9266

[sparsity] Sparsity parametrization (#58705)

z-a-f committed 3 years ago

80cab105

[sparsity][refactor] Changing linear row/col control (#60850)

z-a-f committed 3 years ago

5d34b795

[sparsity] Add sparsity tests to run_test.py (#60887)

z-a-f committed 3 years ago

509b1ef9

Sparse: Remove dispatch in parallel region (#60598)

peterbell10 committed 3 years ago

54673fc9

[DDP] Refactor hook running logic (#61018)

rohan-varma committed 3 years ago

11b722c0

[DDP] Remove SPMD from get_bucket_tensors (#61017)

rohan-varma committed 3 years ago

b21df03f

[JIT] Add `torch._C.ScriptList`` (#52832)

Meghan Lele committed 3 years ago

4a2e8b53

Ignore notebooks when checking for newlines (#61156)

driazati committed 3 years ago

6e9e30cc

[quant][fx][perf] improve runtime of prepare step for large models (#61132)

supriyar committed 3 years ago

a4d86e0d

[DataLoader] Add notebook with DataPipes API example (#60680)

VitalyFedyunin committed 3 years ago

277b310e

[pruner] Make bias hook stateless (#61077)

kazhou committed 3 years ago

ca2702a7

[pruner] Add bias support (#60970)

kazhou committed 3 years ago

0a787523

MAINT Adds test and docs for Linear with no batch dims (#60992)

thomasjpfan committed 3 years ago

87dbdef6

Add aten::avgpool2d NNAPI converter (#58538)

Akshit Khurana committed 3 years ago

369802a5

model_dump working with delegate models (#61043)

iseeyuan committed 3 years ago

19b6ee4d

Improved sparse CSR tensor sampling method (#60283)

pearu committed 3 years ago

374278f4

Make pytorch clang-tidy clean (#60649)

guotuofeng committed 3 years ago

6ecc1a4c

Fix documentation preview instructions (#61080)

driazati committed 3 years ago

a0a9ea65

Update DDP documentation to mention outputs not used in loss is supported (#60275)

rohan-varma committed 3 years ago

60509f89

Fix TSAN issue in distributed tests (#59238)

lw committed 3 years ago

0128eb9a

Expose raw saved tensors for codegen functions (#60565)

Varal7 committed 3 years ago

5b44d817

Newer Older