Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update cudnn convolution kernel #10440

Open
wants to merge 13 commits into
base: master
Choose a base branch
from
Open

Update cudnn convolution kernel #10440

wants to merge 13 commits into from

Commits on Mar 6, 2024

  1. add cudnn-frontend dependency

    @linzs148
    linzs148 committed Mar 6, 2024
    Configuration menu
    Copy the full SHA
    35ff41c View commit details
    Browse the repository at this point in the history

  2. auto format by CI

    @oneflow-ci-bot
    oneflow-ci-bot committed Mar 6, 2024
    Configuration menu
    Copy the full SHA
    dc8a96e View commit details
    Browse the repository at this point in the history

Commits on Mar 17, 2024

  1. add cudnn v8 conv forward kernel

    @linzs148
    linzs148 committed Mar 17, 2024
    Configuration menu
    Copy the full SHA
    6e01476 View commit details
    Browse the repository at this point in the history

Commits on Mar 27, 2024

  1. add cudnn v8 conv backward kernel

    @linzs148
    linzs148 committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    49e2c18 View commit details
    Browse the repository at this point in the history

Commits on Mar 28, 2024

  1. transform output type to fp32 for input type fp16

    @linzs148
    linzs148 committed Mar 28, 2024
    Configuration menu
    Copy the full SHA
    e19d08a View commit details
    Browse the repository at this point in the history

Commits on Apr 10, 2024

  1. refine workspace assign for conv v8

    @linzs148
    linzs148 committed Apr 10, 2024
    Configuration menu
    Copy the full SHA
    cd38272 View commit details
    Browse the repository at this point in the history

  2. fix comment

    @linzs148
    linzs148 committed Apr 10, 2024
    Configuration menu
    Copy the full SHA
    19f46f2 View commit details
    Browse the repository at this point in the history

  3. change heuristic value

    @linzs148
    linzs148 committed Apr 10, 2024
    Configuration menu
    Copy the full SHA
    1422954 View commit details
    Browse the repository at this point in the history

Commits on Apr 24, 2024

  1. move cmake files to external

    @linzs148
    linzs148 committed Apr 24, 2024
    Configuration menu
    Copy the full SHA
    ddeb907 View commit details
    Browse the repository at this point in the history

Commits on Apr 30, 2024

  1. fix cmake bug

    @linzs148
    linzs148 committed Apr 30, 2024
    Configuration menu
    Copy the full SHA
    18625d9 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'master' into feat/cudnn_conv_v8

    @linzs148
    linzs148 committed Apr 30, 2024
    Configuration menu
    Copy the full SHA
    b25ad47 View commit details
    Browse the repository at this point in the history

  3. install cudnn_frontend

    @linzs148
    linzs148 committed Apr 30, 2024
    Configuration menu
    Copy the full SHA
    a8fd7a3 View commit details
    Browse the repository at this point in the history

  4. refine functions

    @linzs148
    linzs148 committed Apr 30, 2024
    Configuration menu
    Copy the full SHA
    bea5080 View commit details
    Browse the repository at this point in the history