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

Fix the execution of ANISymmetryFunctions on a non-default CUDA stream #37

Merged
merged 6 commits into from
Nov 22, 2021
Merged

Fix the execution of ANISymmetryFunctions on a non-default CUDA stream #37

merged 6 commits into from
Nov 22, 2021

Commits on Nov 11, 2021

  1. Test ANISymmetryFunctions at a non-default stream

    Raimondas Galvelis committed Nov 11, 2021
    Configuration menu
    Copy the full SHA
    12b4ae6 View commit details
    Browse the repository at this point in the history

  2. Implement CudaANISymmetryFunctions::setStream

    Raimondas Galvelis committed Nov 11, 2021
    Configuration menu
    Copy the full SHA
    7a8a2ca View commit details
    Browse the repository at this point in the history

  3. Set the CUDA stream in the PyTorch wrapper

    Raimondas Galvelis committed Nov 11, 2021
    Configuration menu
    Copy the full SHA
    bdcd8fc View commit details
    Browse the repository at this point in the history

Commits on Nov 15, 2021

  1. Don't assume the same stream for the forward and backward passes

    Raimondas Galvelis committed Nov 15, 2021
    Configuration menu
    Copy the full SHA
    e866777 View commit details
    Browse the repository at this point in the history

  2. Simplify the code

    Raimondas Galvelis committed Nov 15, 2021
    Configuration menu
    Copy the full SHA
    41fb447 View commit details
    Browse the repository at this point in the history

  3. Implement CudaANISymmetryFunctions::getStream

    Raimondas Galvelis committed Nov 15, 2021
    Configuration menu
    Copy the full SHA
    45b1e05 View commit details
    Browse the repository at this point in the history