strisunshinewentingwang/pytorch

Pull Requests Commits

[FSDP][Easy] `named_parameters()`, `named_buffers()` refactor

awgu committed 2 years ago

fa6e3cf0

[pytorch][require export] Don't require the repo to be cloned

bigfootjon committed 2 years ago

01c8ac3b

[SR] Add a copy variant for fused_split_and_squeeze

mikeiovine committed 2 years ago

98b4a410

ns for fx: reenable tests disabled by #62608

vkuzo committed 2 years ago

63c6209d

ns for fx: fix bug to enable again on torchvision models

vkuzo committed 2 years ago

f1f185f6

ns for fx: fix issue with shadowing nodes of unknown dtype

vkuzo committed 2 years ago

ae321042

[ONNX] Support aminmax

BowenBao committed 2 years ago

f78e0fc9

[ONNX] Support dynamic scale & zero_point for fake_quantize_per_tensor_affine

BowenBao committed 2 years ago

6305e572

update codeowner for public API

albanD committed 2 years ago

d4cce305

Revert "remove fp16 support from cpu linalg functions"

pytorchmergebot committed 2 years ago

495c5aeb

remove fp16 support from cpu linalg functions

ngimel committed 2 years ago

de18c28a

Relanding shape cache (75400) (#75710)

Krovatkin committed 2 years ago

ce842f43

Revert "Relanding shape cache (75400)"

pytorchmergebot committed 2 years ago

db180109

[lint] fixes to mypy linter

suo committed 2 years ago

fe1e6de7

Add checks for public and private API

anjali411 committed 2 years ago

d6e6061b

baby steps on patching inf/nan behavior & aten::amin support in nvfuser

jjsjann123 committed 2 years ago

692ebc8d

Fix: Make `nn.init.orthogonal_` no-op for empty input

jaketae committed 2 years ago

b5b296c4

[JIT] nvfuser test - use only major & minor versions

davidberard98 committed 2 years ago

790cc8f2

Add forward AD codegen support for single formula returning multiple outputs

soulitzer committed 2 years ago

d7b29f3e

Add forward AD support for norm, dist, F.pairwise_dist, F.normalize

soulitzer committed 2 years ago

8721abc4

Revert "Remove histogramdd functional wrapper"

pytorchmergebot committed 2 years ago

3471b0eb

Do not ignore lazy/generated/README.me

malfet committed 2 years ago

51666ff1

fix PostLocalSGDOptimizer and ModelAverager average bug

Haijunlv committed 2 years ago

08f3b958

Relanding shape cache (75400)

Krovatkin committed 2 years ago

89486821

[PyTorch Distributed] Fix batch_isend_irecv

kwen2501 committed 2 years ago

d6f22abb

Expose some functions out of ENABLE_FLATBUFFER.

qihqi committed 2 years ago

3c505fb4

Add `__torch_function__` override protocol supporting to some factory functions

chengjunlu committed 2 years ago

ac8d2201

Remove histogramdd functional wrapper

peterbell10 committed 2 years ago

7c901712

[shard] fix sharded_tensor tests

wanchaol committed 2 years ago

4afe2db6

[quant][fx][fix] Add additional checks when tracing back during maybe share output observer function

jerryzh168 committed 2 years ago

bc371a2c

Newer Older