Как это понять на сайте pytorch?

Question

Я замечаю это на официальном сайте pytorch: https://pytorch.org/docs/stable/nn.html

Если выполняются следующие условия:

1) cudnn включен,

2) входные данные находятся на графическом процессоре,

3) входные данные имеют тип d torch.float16,

4) используется графический процессор V100, а

5) входные данные не в формате PackedSequence.

Затем для повышения производительности можно выбрать постоянный алгоритм.

Может ли кто-нибудь объяснить это? Благодаря.

Answer 1

Это относится к очень низкоуровневой оптимизации производительности использования кэша графического процессора, что более подробно объясняется здесь (примечание: это не материал PyTorch, но я считаю, что он достаточно хорошо работаетобъясняя)Другими словами, если все маркеры удовлетворены, PyTorch по умолчанию будет использовать другой алгоритм под капотом, надеясь обеспечить более высокую производительность RNN.