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.

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

[topi][CuDNN] Removed requirement for GPU from topi conv2d_cudnn.cuda and conv3d_cudnn.cuda #8276

Merged
merged 1 commit into from
Jun 18, 2021

Commits on Jun 17, 2021

  1. [topi][CuDNN] Removed requirement for GPU from topi conv2d_cudnn.cuda…

    … and conv3d_cudnn.cuda
    
    Previously, `conv2d_cudnn.cuda` would use cudnn's benchmarking
    function to select a forward convolution when `cfg.is_fallback`, and
    `conv3d_cudnn.cuda` would use cudnn's benchmarking at all times.
    After this commit, both expose the cudnn algorithm choice as an
    option.  If `cfg.is_fallback`, the local device will be benchmarked if
    present, otherwise will select a default cudnn implementation.
    
    In the future, to better support RPC use-cases, the fallback config
    should be based on cudnn-specific parameters saved in the Target
    object.
    Lunderberg committed Jun 17, 2021
    Configuration menu
    Copy the full SHA
    f7fa507 View commit details
    Browse the repository at this point in the history