mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
4fef3763dd
Summary: Original PR: https://github.com/pytorch/pytorch/pull/37419 cc mattip suo Pull Request resolved: https://github.com/pytorch/pytorch/pull/37778 Differential Revision: D21385774 Pulled By: ezyang fbshipit-source-id: 5de532faab8bae132736b6b5189e0ee2ac9935be
13 lines
321 B
ReStructuredText
13 lines
321 B
ReStructuredText
:orphan:
|
|
|
|
|
|
.. note::
|
|
|
|
If the following conditions are satisfied:
|
|
1) cudnn is enabled,
|
|
2) input data is on the GPU
|
|
3) input data has dtype ``torch.float16``
|
|
4) V100 GPU is used,
|
|
5) input data is not in ``PackedSequence`` format
|
|
persistent algorithm can be selected to improve performance.
|