strisunshinewentingwang/pytorch

Pull Requests Commits

PyTorch model metadata. (#16275)

pritamdamania committed 5 years ago

c3f5ba94

Trim libshm deps, move tempfile.h to c10 (#17019)

Dmytro Dzhulgakov committed 5 years ago

46503a7a

Implement NetDef <--> JIT IR converters. (#16967)

Mikhail Zolotukhin committed 5 years ago

d25fee31

Remove IgnoredPythonOp sugared value

David Riazati committed 5 years ago

decc0893

Separate reduce functions from math (#16929)

xiaomengy committed 5 years ago

3a34f443

Skip test_cudnn_multiple_threads_same_device on ROCm (flaky) (#17061)

bddppq committed 5 years ago

9b7f3da7

Support FC (Caffe2) -> Gemm (ONNX) with variable input shape. (#16184)

xkszltl committed 5 years ago

a670824f

Make timeout in resnet50_trainer configurable (#17058)

bddppq committed 5 years ago

2ad5dcbb

Make mkldnn Stream object thread_local and enable mkldnn thread-safe (#17022)

bdaskalov committed 5 years ago

41dddfd5

Support conversion from Caffe2 MergeDim to ONNX Reshape + Squeeze. (#16189)

xkszltl committed 5 years ago

491f2d4c

Fix mvlgamma doc (#17045)

vishwakftw committed 5 years ago

86594e63

Change IR graph print format to make it look more pythonic (#16986)

Mikhail Zolotukhin committed 5 years ago

f79563a6

Turn off the ability for Declarations.cwrap entries to be methods.

gchanan committed 5 years ago

18b85725

Remove chunk count check on the ChunkBuffer (#16868)

jaliyae committed 5 years ago

bc39cf4d

Use IndexError instead of RuntimeError in ATen CPU kernels

skrah committed 5 years ago

a5e7b1d0

Mark IntList as deprecated; add C10_DEPRECATED_USING (#16824)

ezyang committed 5 years ago

1fc05bd2

Add more debugging facilities to ONNXIFI transform (#17043)

Yinghai Lu committed 5 years ago

db82fc7c

Updating submodules

svcscm committed 5 years ago

26018d02

Updating submodules

svcscm committed 5 years ago

2b5bef22

unify c2 and TH allocator (#16892)

Dmytro Dzhulgakov committed 5 years ago

51dd2000

Updating submodules

svcscm committed 5 years ago

f87022bf

Remove second output of Reshape during ONNXIFI transform (#17027)

Yinghai Lu committed 5 years ago

fb5790ce

enable more unit tests in test_nn (#16994)

iotamudelta committed 5 years ago

9d01be1a

fix bicubic upsampling and enable tests (#17020)

iotamudelta committed 5 years ago

02b838e0

Fold col offsets into bias; optimize A symmetric quant (#16942)

jspark1105 committed 5 years ago

92221ad8

enable unit tests in test_cuda that now pass with ROCm 2.1

iotamudelta committed 5 years ago

3e1e5d5a

Register CUDA kernels for caffe2 operators (#16691)

smessmer committed 5 years ago

9696fee6

Enable test_jit tests that work on ROCm 2.1

iotamudelta committed 5 years ago

059c55f8

Extract ShapeInfo and some util functions into a separate file. (#17025)

ipiszy committed 5 years ago

7c24de8d

Allow customization of blob node in net_drawer (#16915)

Yinghai Lu committed 5 years ago

f435fb82

Newer Older