strisunshinewentingwang/pytorch

Pull Requests Commits

[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

Simplify IfThenElse and CompareSelect within for-loop (#76793)

EikanWang committed 2 years ago

e5a5cd14

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

linbinyu committed 2 years ago

168dc70f

Discover and check operator variants

amjames committed 2 years ago

e5a752a6

Connect Tensor.__ipow__ to pow_ method

amjames committed 2 years ago

39d3a7ff

Fix `mean` bug for integral tensors. (#76584)

ysiraichi committed 2 years ago

459e10f4

[NVFuser] fix force-disable flag

davidberard98 committed 2 years ago

e175065c

Newer Older