summaryrefslogtreecommitdiff
path: root/torch/backends
AgeCommit message (Expand)AuthorFilesLines
2019-04-18Add device-specific cuFFT plan caches (#19300)Tongzhou Wang1-9/+60
2019-03-25Add ability to query if built with CUDA and MKL-DNN. (#18362)Edward Yang2-0/+14
2019-01-31Add torch.backends.openmp.is_available(); fix some cmake messages (#16425)SsnL1-0/+6
2019-01-30Fix the flake8 linterLu Fang1-1/+1
2018-10-31Add support for torch.backends.cudnn.enabled (#13057)David Riazati1-8/+10
2018-10-10Allow linking to backwards-compatible cuDNN at runtime (#12239)sclarkson1-3/+15
2018-09-09Also set stdin to subprocess pipe in FindCUDNN windows popen call (#11435)Matt Dawkins1-1/+1
2018-06-29Move _cudnn_init_dropout_state to TensorOptions and enable cuDNN dropout in C...Peter Goldsborough1-5/+10
2018-06-22Cache cufft plans (#8344)Tongzhou Wang1-0/+39
2018-06-20Add torch.cuda.cudnn_is_available (#8703)Peter Goldsborough1-1/+6
2018-05-10Split libATen.so into libATen_cpu.so and libATen_cuda.so (#7275)Edward Z. Yang1-1/+1
2018-04-17Codemod to update our codebase to 0.4 standard (#6641)Tongzhou Wang1-1/+1
2018-04-12Separate cuda-ness from dtype. (#6470)gchanan1-2/+2
2018-03-19[fft][1 of 3] build system and helpers to support cuFFT and MKL (#5855)Tongzhou Wang1-0/+6
2018-03-09Support native namespace functions with type dispatch. (#5576)gchanan1-1/+1
2018-03-02Port cuDNN RNN dropout state initialization to ATen and make Python c… (#5383)Edward Z. Yang1-15/+8
2018-02-06Use Variable instead of Tensor in Function.forward (#4786)Sam Gross1-3/+8
2018-02-05Port cuDNN RNN bindings to ATen (#4881)Edward Z. Yang1-420/+14
2018-01-08Fix #4492, make it impossible to forget to reset cudnn flags (#4503)Edward Z. Yang1-3/+36
2017-12-21Further improvements to ATen convolution (#4287)Edward Z. Yang1-8/+17
2017-12-14Propagate CuDNN enabled to ATen library. (#4104)Edward Z. Yang1-5/+36
2017-12-13Add rnn args check (#3925)Richard Zou1-10/+0
2017-11-14Fix the missing importpeter1-0/+1
2017-11-11Fix for cuDNN half precision RNN for pre-volta archs (#3613)Christian Sarofeen1-4/+5
2017-11-08Improve Windows Compatibility (for csrc/scripts) (#2941)peterjc1231-1/+20
2017-11-01Added tensor op check for cudnn rnns (#3409)Sean Naren1-0/+8
2017-10-10Deterministic cudnn algorithmsPriya Goyal1-0/+19
2017-07-25Improve memory usage of cuDNN RNN modules (#2179)Adam Paszke1-16/+29
2017-06-11Add a broadcast parameter to copy_, use it in the library in cases where ther...Gregory Chanan1-1/+1
2017-05-19Check cuDNN version at runtime (#1586)Sam Gross1-0/+6
2017-05-18Fix call to Tensor.set_ in rnn.py (#1592)Sam Gross1-2/+2
2017-03-16Use cuDNN and NCCL symbols from _C library (#1017)Sam Gross1-252/+37
2017-03-01Allow backprop through cuDNN RNN in eval modeAdam Paszke2-29/+42
2017-03-01Add support for variable length sequences in RNNs (#873)Adam Paszke2-36/+69
2017-02-17Fix cuDNN dropout desc for multi-gpu (#772)Christian Sarofeen1-3/+4
2017-02-15Update rnn.pybdfhjk1-1/+1
2017-02-14Add some more asserts to cuDNN RNNAdam Paszke1-2/+8
2017-02-14Add tests for non-contiguous inputs and gradientsAdam Paszke1-0/+2
2017-01-31adding cudnn V6 support (#515)ngimel2-18/+54
2017-01-28Remove unnecessary zero_() calls in cuDNN RNNAdam Paszke1-8/+9
2017-01-28Fix non-contiguous grad handling in cuDNN RNNAdam Paszke1-2/+2
2017-01-28[pep8] Fix most lint automatically with autopep8Luke Yeager2-12/+45
2017-01-13add cudnn deb package installation paths to cudnn discovery, add 5.1.10 to lo...ngimel1-1/+1
2017-01-13fix cudnn rnn batch_first with tests (#445)ngimel1-6/+5
2016-12-30Hold CuDNN PRNG state between RNN iterationsAdam Lerer1-3/+22
2016-12-28Use cuDNN for Conv3d and ConvTranspose3d (#359)Sam Gross1-0/+4
2016-12-28Implement comparison and logical operators for tensorsAdam Paszke1-7/+7
2016-12-21change the order of cudnn libssoumith1-1/+1
2016-12-17add cudnn 5.0.5 to supported versions (#321)Sergey Zagoruyko1-1/+1
2016-12-15Make sure that C extension was compiled with cuDNN before using itAdam Paszke1-0/+7