clab/dynet

Add cuDNN support for RNNs

Open

#574 建立於 2017年5月31日

在 GitHub 查看
 (0 留言) (0 反應) (0 負責人)C++ (701 fork)github user discovery
enhancementhelp wanted

倉庫指標

Star
 (3,433 star)
PR 合併指標
 (PR 指標待抓取)

描述

It would be nice to have cuDNN speedup for SimpleRNN, LSTM and GRU.

While cuDNN will likely not be able to accelerate incremental RNN construction, we can special case the .transduce(seq) and .add_inputs(seq) implementations to make use of the acceleration.

貢獻者指南

研究方向
研究 cuDNN RNN API 並將其整合到 DyNet 的 SimpleRNN、LSTM 和 GRU 實作中，重點在 .transduce() 和 .add inputs() 方法。確保與現有 DyNet RNN 抽象相容。
技術棧
cpp
領域
machine learning
議題類型
功能
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
4
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-2 天
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
新近可參與
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
C++CUDAcuDNNDyNet internals
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
20