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

[1.x][FEATURE] CUDA graphs support #19142

Merged
merged 15 commits into from
Sep 19, 2020
Merged

[1.x][FEATURE] CUDA graphs support #19142

merged 15 commits into from
Sep 19, 2020

Commits on Sep 14, 2020

  1. Initial cherry-pick

    @ptrendx
    ptrendx committed Sep 14, 2020
    Configuration menu
    Copy the full SHA
    62d381a View commit details
    Browse the repository at this point in the history

  2. Store NodeAttrs in OpExecutor

    @ptrendx
    ptrendx committed Sep 14, 2020
    Configuration menu
    Copy the full SHA
    c15357a View commit details
    Browse the repository at this point in the history

  3. Do not allow stateful operations in CUDA graphs and provide mechanism 

    for marking ops as safe
    @ptrendx
    ptrendx committed Sep 14, 2020
    Configuration menu
    Copy the full SHA
    3945895 View commit details
    Browse the repository at this point in the history

  4. Guard against using ops with synchronization

    @ptrendx
    ptrendx committed Sep 14, 2020
    Configuration menu
    Copy the full SHA
    599986a View commit details
    Browse the repository at this point in the history

  5. Cleaning

    @ptrendx
    ptrendx committed Sep 14, 2020
    Configuration menu
    Copy the full SHA
    f7aa654 View commit details
    Browse the repository at this point in the history

  6. Properly guard graphs

    @ptrendx
    ptrendx committed Sep 14, 2020
    Configuration menu
    Copy the full SHA
    5d50710 View commit details
    Browse the repository at this point in the history

  7. Limit graphs to CUDA 10.2+

    @ptrendx
    ptrendx committed Sep 14, 2020
    Configuration menu
    Copy the full SHA
    bb8740e View commit details
    Browse the repository at this point in the history

Commits on Sep 15, 2020

  1. Fix the compilation when graphs are not available

    @ptrendx
    ptrendx committed Sep 15, 2020
    Configuration menu
    Copy the full SHA
    1dcb30e View commit details
    Browse the repository at this point in the history

  2. Guarding the libcuda.so usage behind RTC compilation flag

    @ptrendx
    ptrendx committed Sep 15, 2020
    Configuration menu
    Copy the full SHA
    3746543 View commit details
    Browse the repository at this point in the history

Commits on Sep 16, 2020

  1. Document the env variables

    @ptrendx
    ptrendx committed Sep 16, 2020
    Configuration menu
    Copy the full SHA
    a925ceb View commit details
    Browse the repository at this point in the history

Commits on Sep 17, 2020

  1. Add test

    @ptrendx
    ptrendx committed Sep 17, 2020
    Configuration menu
    Copy the full SHA
    ae4ef9a View commit details
    Browse the repository at this point in the history

  2. Merge branch 'v1.x' into pr_cuda_graphs_1.x

    @ptrendx
    ptrendx committed Sep 17, 2020
    Configuration menu
    Copy the full SHA
    5409163 View commit details
    Browse the repository at this point in the history

Commits on Sep 18, 2020

  1. Fix the test

    @ptrendx
    ptrendx committed Sep 18, 2020
    Configuration menu
    Copy the full SHA
    adae98a View commit details
    Browse the repository at this point in the history

  2. Merge branch 'v1.x' into pr_cuda_graphs_1.x

    @ptrendx
    ptrendx committed Sep 18, 2020
    Configuration menu
    Copy the full SHA
    cfb326c View commit details
    Browse the repository at this point in the history

  3. Use with_environment