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

Add onnx export cuda support #17183

Merged
merged 20 commits into from
May 18, 2022
Merged

Add onnx export cuda support #17183

merged 20 commits into from
May 18, 2022

Commits on May 11, 2022

  1. Add CUDA support

    @JingyaHuang
    JingyaHuang committed May 11, 2022
    Configuration menu
    Copy the full SHA
    126d7e3 View commit details
    Browse the repository at this point in the history

  2. Raise if TF

    @JingyaHuang
    JingyaHuang committed May 11, 2022
    Configuration menu
    Copy the full SHA
    296d590 View commit details
    Browse the repository at this point in the history

  3. Add test for PyTorch export on CUDA

    @JingyaHuang
    JingyaHuang committed May 11, 2022
    Configuration menu
    Copy the full SHA
    756e5b6 View commit details
    Browse the repository at this point in the history

  4. Fix NameError

    @JingyaHuang
    JingyaHuang committed May 11, 2022
    Configuration menu
    Copy the full SHA
    f1c9dd4 View commit details
    Browse the repository at this point in the history

Commits on May 12, 2022

  1. Fix typo 

    Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
    @JingyaHuang @LysandreJik
    JingyaHuang and LysandreJik authored May 12, 2022
    Configuration menu
    Copy the full SHA
    6a96f3a View commit details
    Browse the repository at this point in the history

  2. Fix typo 

    Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
    @JingyaHuang @LysandreJik
    JingyaHuang and LysandreJik authored May 12, 2022
    Configuration menu
    Copy the full SHA
    3c1813c View commit details
    Browse the repository at this point in the history

Commits on May 13, 2022

  1. Update src/transformers/onnx/convert.py 

    Co-authored-by: lewtun <lewis.c.tunstall@gmail.com>
    @JingyaHuang @lewtun
    JingyaHuang and lewtun authored May 13, 2022
    Configuration menu
    Copy the full SHA
    f105ad2 View commit details
    Browse the repository at this point in the history

Commits on May 15, 2022

  1. Change arg from cuda to device

    @JingyaHuang
    JingyaHuang committed May 15, 2022
    Configuration menu
    Copy the full SHA
    898d827 View commit details
    Browse the repository at this point in the history

  2. Resolve conflicts

    @JingyaHuang
    JingyaHuang committed May 15, 2022
    Configuration menu
    Copy the full SHA
    6906db3 View commit details
    Browse the repository at this point in the history

  3. fix style

    @JingyaHuang
    JingyaHuang committed May 15, 2022
    Configuration menu
    Copy the full SHA
    796c67d View commit details
    Browse the repository at this point in the history

Commits on May 16, 2022

  1. fix test

    @JingyaHuang
    JingyaHuang committed May 16, 2022
    Configuration menu
    Copy the full SHA
    dde6481 View commit details
    Browse the repository at this point in the history

  2. Improve docstring

    @JingyaHuang
    JingyaHuang committed May 16, 2022
    Configuration menu
    Copy the full SHA
    904ab89 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'main' into jingya/onnx-export-cuda

    @JingyaHuang
    JingyaHuang committed May 16, 2022
    Configuration menu
    Copy the full SHA
    564cc11 View commit details
    Browse the repository at this point in the history

  4. Fix style

    @JingyaHuang
    JingyaHuang committed May 16, 2022
    Configuration menu
    Copy the full SHA
    8dbf073 View commit details
    Browse the repository at this point in the history

Commits on May 17, 2022

  1. Merge branch 'main' into jingya/onnx-export-cuda

    @JingyaHuang
    JingyaHuang committed May 17, 2022
    Configuration menu
    Copy the full SHA
    ff34c30 View commit details
    Browse the repository at this point in the history

  2. Fix bug in Big Bird

    @JingyaHuang
    JingyaHuang committed May 17, 2022
    Configuration menu
    Copy the full SHA
    3a4833d View commit details
    Browse the repository at this point in the history

  3. Make fix-copies

    @JingyaHuang
    JingyaHuang committed May 17, 2022
    Configuration menu
    Copy the full SHA
    4413ee0 View commit details
    Browse the repository at this point in the history

Commits on May 18, 2022

  1. Revert notebook style

    @JingyaHuang
    JingyaHuang committed May 18, 2022
    Configuration menu
    Copy the full SHA
    d4137b4 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into jingya/onnx-export-cuda

    @JingyaHuang
    JingyaHuang committed May 18, 2022
    Configuration menu
    Copy the full SHA
    ee48136 View commit details
    Browse the repository at this point in the history

  3. revert notebooks

    @JingyaHuang
    JingyaHuang committed May 18, 2022
    Configuration menu
    Copy the full SHA
    fc561fa View commit details
    Browse the repository at this point in the history