strisunshinewentingwang/pytorch

Pull Requests Commits

[quant] Input-Weight Equalization - selective equalization (#61916)

angelayi committed 3 years ago

836b2431

[BF16] Add BF16 support to _aminmax and _anminmax_all operators (#62767)

yusuo committed 3 years ago

e6ef8700

[vulkan] Add _reshape_alias (#62858)

SS-JIA committed 3 years ago

56ff9963

[vulkan] Throw an exception if device does not support Vulkan (#62859)

SS-JIA committed 3 years ago

5f4207eb

Introducing DataChunk for DataPipes batching (#62768)

VitalyFedyunin committed 3 years ago

d3bdf345

Add getPyInterpreter() API (#62659)

ezyang committed 3 years ago

5e5de75f

fix docstring default value of `last_epoch` for SWALR in torch/optim/… (#62799)

cneyang committed 3 years ago

27135f86

Update on "[Reland][DDP] Support not all outputs used in loss calculation"

rohan-varma committed 3 years ago

49668709

Update base for Update on "[Reland][DDP] Support not all outputs used in loss calculation"

rohan-varma committed 3 years ago

7777f310

rename namespace f4d to velox (#61)

mshang816 committed 3 years ago

9573e7a6

[torchelastic][multiprocessing] Print warning message only when child processes are stuck (#62823)

aivanou committed 3 years ago

e1f81c93

Allow FractionalMaxPool 2D and 3D layers to accept 0 dim batch size tensors. (#62083)

v0dro committed 3 years ago

f6c7081a

Add tutorial link (#62785)

awgu committed 3 years ago

8aa12cbf

[opinfo] nn.functional.unfold (#62705)

kshitij12345 committed 3 years ago

64c54f92

[DDP] log gradient ready order and bucket indices (#62751)

rohan-varma committed 3 years ago

9ac56ef0

[DDP] Allow tuning of first bucket (#62748)

rohan-varma committed 3 years ago

80091cb0

OpInfo for `adaptive_avg_pool2d` (#62704)

krshrimali committed 3 years ago

5c431981

Bump protobuf version in CircleCI docker images (#62441)

neginraoof committed 3 years ago

eaaceea8

[jit] Better checking for overload function declarations. (#59956)

zhxchen17 committed 3 years ago

e62189ad

Add batched model to torchdeploy examples (#62836)

wconstab committed 3 years ago

63fa53d3

test, fix sparse * dense exceptions and corner case (#61723)

mattip committed 3 years ago

c8eda919

Simplify hardswish ONNX export graph. (#60080)

PeterL1n committed 3 years ago

8d7786ad

add `OpInfo` for `torch.nn.functional.grid_sample` (#62311)

pmeier committed 3 years ago

7630f407

OpInfo for `nn.functional.avg_pool2d` (#62455)

krshrimali committed 3 years ago

5dbcd563

Preserve memory layout when aten batchnorm is used (#62773)

eqy committed 3 years ago

878943c6

[pruner] generalize bias hook for conv2d (#62430)

kazhou committed 3 years ago

d4529161

ns for fx: add ref_node_target_type (#62685)

vkuzo committed 3 years ago

b524a110

Allow disabled tests to be re-enabled with IGNORE_DISABLED_ISSUES (#62686)

janeyx99 committed 3 years ago

b96acb75

Revert D30094460: [profiler] Re-enable test on Windows

malfet committed 3 years ago

24a26813

Revert D30095246: [pytorch][PR] Enable ncclAvg for reductions

pbelevich committed 3 years ago

0c8ed042

Newer Older