strisunshinewentingwang/pytorch

Pull Requests Commits

Added CUDA support for torch.orgqr (#51348)

IvanYashchuk committed 3 years ago

e29d8477

[FX] Added docstring for concrete_args (#53151)

Chillee committed 3 years ago

0819d5f9

[shape inference] fix pruning

ChunliF committed 3 years ago

e1e19a71

[caffe2] Fix shape inference for pruning ops (#53082)

ChunliF committed 3 years ago

5c1c8cb9

blas copy and axpy to aten (#52345)

aocsa committed 3 years ago

0dac7d86

[nnc] Test cases for uneven split + reorder (#53091)

bertmaher committed 3 years ago

565d8235

use device methods (#52899)

cyyever committed 3 years ago

d8730194

[TensorExpr] IRVerifier: add index verifier for Store. (#53137)

Mikhail Zolotukhin committed 3 years ago

aba33b00

Fix constexpr __host__ warning (#52702)

r-barnes committed 3 years ago

0f7f600e

Implements `torch.linalg.lstsq` (#49093)

nikitaved committed 3 years ago

3ac90132

[StaticRuntime] Clean up (#53096)

Hao Lu committed 3 years ago

c0b31a5b

Fixed out= variant of linalg.inv (#51977)

IvanYashchuk committed 3 years ago

870bac13

enable coverage test for dataloader on Windows (#52550)

mszhanyi committed 3 years ago

fd582af0

Huber loss (#50553)

jbschlosser committed 3 years ago

e86476f7

avoid TLS in has_names (#53003)

Basil Hosmer committed 3 years ago

2c8f9aec

FIX Validates target in cosine_embedding (#53110)

thomasjpfan committed 3 years ago

e2ecfb60

Revert D26720919: [Gradient Compression] Remove some low-level methods of GradBucket class

osalpekar committed 3 years ago

593b0fba

Suppress unsigned comparison warning (#52653)

r-barnes committed 3 years ago

c4c20a5d

Update torch.nn.quantizable.MultiHeadAttention docstring (#53106)

nihui committed 3 years ago

6ab3a8b6

Codegen python bindings to access attributes of grad_fn (#52451)

soulitzer committed 3 years ago

a3a21504

Back out "Revert D26753571: [pytorch][PR] add submodules to sys.modules so their attributes can be pickled" (#53127)

ezyang committed 3 years ago

baed2cfe

[Gradient Compression] Remove some low-level methods of GradBucket class (#53098)

Yi Wang committed 3 years ago

521e1e83

[Gradient Compression] Add is_the_last_bucket_to_allreduce method to GradBucket class (#53010)

Yi Wang committed 3 years ago

b05dd931

[Gradient Compression] Don't provide default values in GradBucket constructor (#53102)

Yi Wang committed 3 years ago

4997c38a

[Gradient Compression] Add get_per_parameter_tensors method to GradBucket class (#53009)

Yi Wang committed 3 years ago

ecb5ac90

Add default arguments to cuda stream and events (#53025)

nikithamalgifb committed 3 years ago

ab7f6f3f

Add wait_for_worker param to TCPStore and fix port in use flaky test failures (#52888)

H-Huang committed 3 years ago

2444b4d1

Store coverage files as artifacts for better debugging (#53126)

janeyx99 committed 3 years ago

41765d46

Add a note in DDP doc to point to ZeroRedundancyOptimizer (#53113)

mrshenli committed 3 years ago

d6970902

[Reland] Update and expose ZeroRedundancyOptimizer docs (#53112)

mrshenli committed 3 years ago

29034b94

Newer Older