strisunshinewentingwang/pytorch

Pull Requests Commits

Revert "enable NVFuser by default"

pytorchmergebot committed 2 years ago

2a905aef

skip primTorch nvfuser tests on rocm (#77468)

jjsjann123 committed 2 years ago

8875453d

[Model Averaging] Support disabling post-local gradient sync (#76723)

wayi1 committed 2 years ago

5ab8afe4

[coreml] Use special throw macro when encountering CoreML API errors (#77429)

SS-JIA committed 2 years ago

93d84c0f

[metal] Use special throw macro when encountering Metal API errors (#77430)

SS-JIA committed 2 years ago

c34be4eb

[GHA] Use silent checkout (#77547)

malfet committed 2 years ago

46e16737

[chalf] exp, log: cuda (#77483)

kshitij12345 committed 2 years ago

a8261b4b

[metal] fix op registration for cat (#77451)

SS-JIA committed 2 years ago

9c94b760

Adding maximize to Adamax (#77409)

zaxtax committed 2 years ago

2a496e2f

[Profiler] Remove `addMemoryUsageActivity` from kineto_shim (#76678)

Taylor Robie committed 2 years ago

c3c92ca1

[BE][CI] Add `pip_install` macro

malfet committed 2 years ago

1db13374

Python function to extract information on mobile::Module from flatbuffer (#77328)

qihqi committed 2 years ago

69fa49f1

Fix optional dtype/layout/memory_format pycall; fix memory format

ezyang committed 2 years ago

b5bc954a

Saving JIT to flatbuffer should respect options. (#77456)

qihqi committed 2 years ago

14e59edd

Disable GC if fraction is not set (#76648)

jaewonlee-fb committed 2 years ago

3e89a1d6

add BFloat16 support for BatchNorm on CPU

mingfeima committed 2 years ago

8c504142

Revert "[mobile] add buck build for mobile targets (#76480)"

pytorchmergebot committed 2 years ago

530481ed

[FSDP] Sharded Grad Scaler (#76918)

Sisil Mehta committed 2 years ago

9d3ffed3

Update forward AD metadata check to skip stride check when size is 0

soulitzer committed 2 years ago

beb40503

[LTC] Mark Step Indicator (#76840)

antoniojkim committed 2 years ago

c2182632

[chalf] abs, sgn (#77446)

kshitij12345 committed 2 years ago

3feb0384

missing header include in SortImpl.cu (#77297)

jeffdaily committed 2 years ago

3c804989

[ONNX] Refactor to remove inline imports - attempt 2 (#77448)

justinchuby committed 2 years ago

563c2719

Pretty-print dataclasses (#76810)

francescocastelli committed 2 years ago

dca416b5

add native kernel for weight_norm on CPU

mingfeima committed 2 years ago

6fa20bdf

[FSDP] Add unittests to test distributed checkpointing

fegin committed 2 years ago

a0cc38ea

Add ccol_indices and row_indices methods.

pearu committed 2 years ago

88205886

[DataLoader] Adding ability to use dill to pass DataPipes in mutiprocessing

VitalyFedyunin committed 2 years ago

edffd595

[FSDP] Implement sharded_state_dict and load_sharded_state_dict

fegin committed 2 years ago

58c9d521

[PT-D][Sharding] Enable more ops needed in the transformer model training

fduwjj committed 2 years ago

a2cb94d2

Older