strisunshinewentingwang/pytorch

Pull Requests Commits

Move aten::__contains__.int_list for lite jit (#48950)

jiatongzhou committed 3 years ago

c3a90bed

docker: Add make variable to add docker build args (#48942)

seemethere committed 3 years ago

881e9583

CUDA BF16 backwards (#48809)

zasdfgbnm committed 3 years ago

5533be51

[DOCS]Correct docs for torch.lu_solve (#47762)

bharatr21 committed 3 years ago

3aeb9cc8

Added entry for torch.linalg.cond to linalg.rst (#48941)

IvanYashchuk committed 3 years ago

bea88ee1

[Gradient Compression] Let the dtype of created low-rank tensors P and Q be the same type as the input tensor (#48902)

Yi Wang committed 3 years ago

c876d4f4

Register OpInfos for torch.fft transforms (#48427)

peterbell10 committed 3 years ago

533c8378

[package] pre-emptively install submodules (#48799)

zdevito committed 3 years ago

adbb74de

Reenable some BF16 tests on CUDA (#48805)

zasdfgbnm committed 3 years ago

e3893b86

Update torch.randint documentation to include missing note (#48787)

H-Huang committed 3 years ago

7629612f

Fix CI by removing gen_pyi from mypy-stirct.ini (#48961)

malfet committed 3 years ago

f67259fe

[Docs] Add examples for new object-based c10d APIs (#43932)

rohan-varma committed 3 years ago

b77ca9e8

Add object-based collective APIs to public docs (#48909)

rohan-varma committed 3 years ago

d6b5f3ad

Revert D25304229: [pytorch][PR] Add type annotations to torch.onnx.* modules

ezyang committed 3 years ago

88ebf6f8

Revert D24923679: Fixed einsum compatibility/performance issues (#46398)

heitorschueroff committed 3 years ago

d3076013

#48733 added logging statements to LLVM codegen using JIT logging (#48758)

huiguoo committed 3 years ago

924b001b

[WIP] Added foreach_trunc, foreahc_reciprocal, foreach_sigmoid APIs (#47385)

Iurii Zdebskyi committed 3 years ago

dad74e58

pyi codegen update - remove Declarations.yaml (#48754)

bdhirsh committed 3 years ago

ba6511b3

Fix generator exhaustion in SparseAdam (#47724)

mariosasko committed 3 years ago

f2c3efd5

[vulkan] test_app for mobilenetV2 on vulkan api (#48924)

IvanKobzarev committed 3 years ago

21ba48fe

Fix incorrect usage of CUDACachingAllocator [v2] (#48817)

xwang233 committed 3 years ago

36df2533

Add type annotations to torch.onnx.* modules (#48782)

guilhermeleobas committed 3 years ago

8bc6023d

Add explicit cast to cuda_atomic_ops_test.cu (#48886)

malfet committed 3 years ago

1febd222

[Caffe2]Add more error message in ComputeBinaryBroadcastForwardDims

Meng Zhang committed 3 years ago

00f01791

CUDA BF16 norm (#48806)

zasdfgbnm committed 3 years ago

a39398b9

Add another torch::jit::load API to load PyTorch model with shared_ptr PyTorchStreamReader input (#48802)

lliu315gt committed 3 years ago

19f4c511

Fixing error: "member may not be initialized" due to constexpr at Windows (#48836)

datumbox committed 3 years ago

e429d050

Fixed einsum compatibility/performance issues (#46398) (#47860)

heitorschueroff committed 3 years ago

ea2a568c

[Gradient Compression] Replace the key of error_dict in PowerSGD state with bucket index (#48867)

Yi Wang committed 3 years ago

17f53bff

[numpy] `torch.sinh`: promote integer inputs to float (#48644)

kshitij12345 committed 3 years ago

2e600fed

Newer Older