strisunshinewentingwang/pytorch

Pull Requests Commits

[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

Condense JIT/Quantization triage into one workflow (#61130)

samestep committed 3 years ago

3f0f860a

Use a leaky singleton for CublasHandlePool. (#60987)

pritamdamania committed 3 years ago

6f92f10c

add embedding bag skeleton take 2 (#61126)

Hector Yuen committed 3 years ago

d2fef350

[jit] Allow instance overrides of ignored methods (#61076)

suo committed 3 years ago

e5ae0e65

Revert D29413019: [torch] Various improvements to `torch.distributed.launch` and `torch.distributed.run`

VitalyFedyunin committed 3 years ago

ccfdb306

[DataLoader] Add Example Only `fork` DataPipe (#60679)

VitalyFedyunin committed 3 years ago

48bfc0e5

[DataLoader] Decorate ZipDataPipe as `zip` (#60678)

VitalyFedyunin committed 3 years ago

62b2dc20

[DataLoader] Add simple `groupby` DataPipe (#60675)

VitalyFedyunin committed 3 years ago

8e21ff91

Revert D28836794: SumKernel (BFloat16): use float as accumulation type

VitalyFedyunin committed 3 years ago

cb7d8132

Fix some integer comparisons (#60894)

r-barnes committed 3 years ago

11dca2e5

Revert D29313058: add an embedding bag skeleton operators

Rong Rong (AI Infra) committed 3 years ago

7017dc10

Newer Older