Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[WIP][MXNET-107] Fused LSTM implementation for CPU #10104

Merged
merged 39 commits into from
May 14, 2018
Merged

[WIP][MXNET-107] Fused LSTM implementation for CPU #10104

merged 39 commits into from
May 14, 2018

Commits on May 5, 2018

  1. register RNN fused-API with nnvm, finish single-layer && undirection … 

    …LSTM forward function
    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    fd24ed2 View commit details
    Browse the repository at this point in the history

  2. fix coding style and lint complains

    @TaoLv
    TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    ba0fe6d View commit details
    Browse the repository at this point in the history

  3. add single-layer && undirectional LSTM backward function

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    a3c34ab View commit details
    Browse the repository at this point in the history

  4. make interface universal for other RNN mode

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    b5c1ef7 View commit details
    Browse the repository at this point in the history

  5. share intermediate result between forward and backward in a trick way

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    73ed6dd View commit details
    Browse the repository at this point in the history

  6. add comments for important parameters

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    d72fe17 View commit details
    Browse the repository at this point in the history

  7. modify testcase

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    d6811b5 View commit details
    Browse the repository at this point in the history

  8. Fix coding style and error message

    @TaoLv
    TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    d0306e5 View commit details
    Browse the repository at this point in the history

  9. fix openmp collapse error

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    c2e7c8f View commit details
    Browse the repository at this point in the history

  10. fix const

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    154aa3b View commit details
    Browse the repository at this point in the history

  11. remove rnn.cu and skip related testcases temporarily for building on GPU

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    7c0cc29 View commit details
    Browse the repository at this point in the history

  12. support multi-layer and bidirectional for lstm inference

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    b59f009 View commit details
    Browse the repository at this point in the history

  13. remove some testcaseS in test_gluon_rnn.py to build on GPU

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    26d32d2 View commit details
    Browse the repository at this point in the history

  14. remove testcase between fp32 and fp64 temporarily

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    1b89cff View commit details
    Browse the repository at this point in the history

  15. retrigger ci

    @TaoLv
    TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    afd831d View commit details
    Browse the repository at this point in the history

  16. fix some logs

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    ce818d3 View commit details
    Browse the repository at this point in the history

  17. use a better way to share memory

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    f24ee4b View commit details
    Browse the repository at this point in the history

  18. fix cudnn registration

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    d51dafd View commit details
    Browse the repository at this point in the history

  19. fix invariant calculations and enable some gpu testcases

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    cdaadf7 View commit details
    Browse the repository at this point in the history

  20. add thread local cache for cudnn rnn op

    @TaoLv
    TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    4161f3b View commit details
    Browse the repository at this point in the history

  21. add thread local cache for rnn op

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    f3dcb07 View commit details
    Browse the repository at this point in the history

  22. fix bugs

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    09f6e9a View commit details
    Browse the repository at this point in the history

  23. remove some testcases to check segmentfault

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    c28bbc8 View commit details
    Browse the repository at this point in the history

  24. remove cudnn registeration to check segmentfault

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    3370cb4 View commit details
    Browse the repository at this point in the history

  25. support multi-layer for LSTM Training

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    46af847 View commit details
    Browse the repository at this point in the history

  26. modify lstm testcase

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    e42e7f9 View commit details
    Browse the repository at this point in the history

  27. add bidirectional support for lstm

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    e5b8b51 View commit details
    Browse the repository at this point in the history

  28. fix gluon and coding style

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    8a67315 View commit details
    Browse the repository at this point in the history

  29. fix bugs

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    78edb41 View commit details
    Browse the repository at this point in the history

  30. remove nnvm registration

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    f50f5c0 View commit details
    Browse the repository at this point in the history

  31. enable gpu testcases

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    35a4a4b View commit details
    Browse the repository at this point in the history

  32. add detailed descriptions

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    19ef217 View commit details
    Browse the repository at this point in the history

  33. add dropout check

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    b0cfcf8 View commit details
    Browse the repository at this point in the history

  34. fix workspace size

    @TaoLv
    zhangshu authored and TaoLv committed May 5, 2018
    Configuration menu
    Copy the full SHA
    b6b567e View commit details
    Browse the repository at this point in the history

Commits on May 8, 2018

  1. Merge remote-tracking branch 'upstream/master' into lstm

    @TaoLv
    TaoLv committed May 8, 2018
    Configuration menu
    Copy the full SHA
    1471836 View commit details
    Browse the repository at this point in the history

Commits on May 9, 2018

  1. dropout is not supported, add unit test for it

    @TaoLv
    TaoLv committed May 9, 2018
    Configuration menu
    Copy the full SHA
    a52b5ef View commit details
    Browse the repository at this point in the history

  2. Merge remote-tracking branch 'upstream/master' into lstm

    @TaoLv
    TaoLv committed May 9, 2018
    Configuration menu
    Copy the full SHA
    a60de72 View commit details
    Browse the repository at this point in the history

Commits on May 12, 2018

  1. fix review comments

    @TaoLv
    TaoLv committed May 12, 2018
    Configuration menu
    Copy the full SHA
    3c61b84 View commit details
    Browse the repository at this point in the history

  2. Merge remote-tracking branch 'upstream/master' into lstm

    @TaoLv
    TaoLv committed May 12, 2018
    Configuration menu
    Copy the full SHA
    aeb8e9d View commit details
    Browse the repository at this point in the history