strisunshinewentingwang/pytorch

Pull Requests Commits

add potrf_buffersize, use loop of potrf for batch

xwang233 committed 3 years ago

faa14809

[Gradient Compression] Make FP16 compression as a wrapper that can be combined with other communication hooks (#53808)

iseessel committed 3 years ago

3078233e

[caffe2] Don't call TensorImpl::size() in dim32() (#53852)

swolchok committed 3 years ago

8a5b946f

xwang233 committed 3 years ago

e267a475

rename lib, legacy to cusolver, magma

xwang233 committed 3 years ago

a78a207f

use get_device_pointers

xwang233 committed 3 years ago

7f6294ab

Merge remote-tracking branch 'upstream/master' into cusolver-cholesky-1

xwang233 committed 3 years ago

c63e1d99

Revert D26922420: [ONNX] fix export of embedding with padding_idx (#53053)

malfet committed 3 years ago

5b648ef9

[reland] Add OpInfo for `bitwise_not` (#53181)

kshitij12345 committed 3 years ago

00771eff

Added cuBLAS path for torch.triangular_solve (#53147)

IvanYashchuk committed 3 years ago

fe086717

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

nikitaved committed 3 years ago

afa1ff8e

[Static Runtime] Fix bug in ClipRangesGatherRangesX2SigridHash (#53799)

Hao Lu committed 3 years ago

49323423

Revert D26993790: [pytorch][PR] [CUDA graphs] Private mempools for CUDA graphs

ngimel committed 3 years ago

76129c7c

[fix] torch.cat : cross-device check for out and input tensors (#53004)

kshitij12345 committed 3 years ago

fe38027f

compareSet method for HashStore and FileStore (#53803)

siva1007 committed 3 years ago

fdbd667e

Fix typo "informations" -> "information" (#53746)

lithuak committed 3 years ago

d4c877b5

Add missing decorators in test_spectral_ops (#53736)

peterbell10 committed 3 years ago

f62e9156

fix for method_tests() random failures (#53854)

aocsa committed 3 years ago

89fce74d

[caffe2] Support deserializing tensors using alternate serialization formats (#53403)

simpkins committed 3 years ago

33aaea91

[caffe2] add a CAFFE2_NODISCARD macro to help support old compilers (#53754)

simpkins committed 3 years ago

91531d30

Use the conda channel defined in docker.Makefile to install cudatoolkit (#53316)

xuzhao9 committed 3 years ago

7763bb6c

[CUDA graphs] Private mempools for CUDA graphs (#51436)

mcarilli committed 3 years ago

90dfdef2

[PyTorch] Remove unnecessary assert in maybe_resize_storage_cpu (#53724)

swolchok committed 3 years ago

804f3f98

Replace thrust with cub in randperm (#53841)

zasdfgbnm committed 3 years ago

34eb644e

Skip dispatch for is_signed (#53847)

Caroline Chen committed 3 years ago

7f4aff82

ns for fx: move linear activation test case to new API (#53777)

vkuzo committed 3 years ago

2912ad13

ns for fx: move compare activations for conv test to new API (#53776)

vkuzo committed 3 years ago

57bf1340

ns for fx: move lstm dynamic weight test case to new API (#53772)

vkuzo committed 3 years ago

01c6e936

ns for fx: move linear dynamic weight test case to new API (#53765)

vkuzo committed 3 years ago

a71cd135

ns for fx: move linear weight test case to new API (#53764)

vkuzo committed 3 years ago

9c8f112a

Newer Older