strisunshinewentingwang/pytorch

Pull Requests Commits

[DataLoader] Change signature of Functional DataPipe (#52458)

ejguan committed 3 years ago

4ee5bc74

Enable min & max for Float16 & BFloat16 (#51244)

imaginary-person committed 3 years ago

3adc8f8c

[quant][graphmode][fx] Fix fp16 dynamic quant for functional linear (#52369)

jerryzh168 committed 3 years ago

fb9f8950

Add support for pow (#52374)

nikithamalgi committed 3 years ago

83feaebf

Add NNC support for aten::hardtanh (a hot operation in mobilenet v2/v3) (#52394)

huiguoo committed 3 years ago

d8b28579

Add onnxifi interface for set/get options (#52388)

Yinghai Lu committed 3 years ago

f4c33edb

[ROCm] missing template declarations for complex blas (#52472)

jeffdaily committed 3 years ago

82548f3a

Improvements for FX tracer (#52232)

jfix71 committed 3 years ago

65f6e665

[glow] Extending AOT config with two more fields (#5359)

khabinov committed 3 years ago

bb7e07ce

[ROCm] Enable test_ddp_hooks.py test cases (#52403)

KyleCZH committed 3 years ago

e0b6252d

Add arm64 binary build (#52443)

janeyx99 committed 3 years ago

89bc9a58

Revert D26299594: [PyTorch Mobile] 15KiB size reduction by reducing MaxTypeIndex from 256 to 32

malfet committed 3 years ago

22adea04

Revert nightly docker build cuda version to 11.1.1. (#52234)

xuzhao9 committed 3 years ago

2d435442

[iOS GPU] Fix max_pool_2d (#52431)

xta0 committed 3 years ago

49c90648

Make LLVM the default backend for TE (#52314)

navahgar committed 3 years ago

c7a70eec

enable mkldnn conv2d backward to support mkldnn tensor input (#48994)

XiaobingSuper committed 3 years ago

8f3ed60d

[PyTorch Mobile] 15KiB size reduction by reducing MaxTypeIndex from 256 to 32 (#51881)

dhruvbird committed 3 years ago

9e545329

Allow broadcasting against lerp weights. (#52319)

gchanan committed 3 years ago

983347fa

[BE] _get_torch_cuda_version should return tuple (#52409)

Rong Rong (AI Infra) committed 3 years ago

b52e2e60

Fix upsample bicubic2d batching handling on CPU. (#52389)

gchanan committed 3 years ago

f72b4b83

Enhance Tensor.unflatten to support -1 as the inferred size (#51955)

RockingJavaBean committed 3 years ago

c7b00058

ns for fx: make unshadowed activation comparison work for N models (#52357)

vkuzo committed 3 years ago

ad974645

ns for fx: make weights comparison work on N models (#52356)

vkuzo committed 3 years ago

a937d1cb

[wip] ns for fx: add support for subgraph matching (#52130)

vkuzo committed 3 years ago

d903106b

NS for FX: add test for a simple sparsenn model (#52092)

vkuzo committed 3 years ago

3978ffb3

[PyTorch] Avoid std::string in TORCH_CHECK when possible (#52221)

swolchok committed 3 years ago

efbb854e

[PyTorch][easy] Make shared empty string static instead of thread_local (#52220)

swolchok committed 3 years ago

ba77b8d8

Make bias in lazy modules lazy and avoid create empty tensors (#52212)

zhuzilin committed 3 years ago

c8b3686a

Revert D26369476: [pytorch][PR] [complex] `masked_fill`: Complex Autograd support and update masked_scatter skips.

anjali411 committed 3 years ago

758aa455

[deploy] torch::deploy API (#51754)

zdevito committed 3 years ago

60518d10

Newer Older