Add NWH + B2B causal conv1d notebooks; refresh outputs#246
Open
yeliu-oss wants to merge 1 commit into
Open
Conversation
- Add 62_causal_conv1d_nwh_forward.ipynb - Add 63_causal_conv1d_nwh_backward.ipynb - Add 64_b2b_causal_conv1d_forward.ipynb - Add 65_b2b_causal_conv1d_backward.ipynb - Refresh outputs for all 6 notebooks (60-65)
Member
|
@cudnn-ci-bot run |
|
Backend pipeline not launched Reason: PR base branch 'develop' does not match 'main' |
Member
|
@cudnn-ci-bot run |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Description
Add causal conv1d support to cuDNN frontend: Python bindings, C++ samples, and Jupyter notebooks for all 6 kernel variants (NHW fwd/bwd, NWH fwd/bwd, B2B fwd/bwd).
Changes Made
Python bindings (
python/cudnn/ops/causal_conv1d.py):cudnn.ops.causal_conv1d— NHW layout (fwd + bwd via autograd)cudnn.ops.causal_conv1d_nwh— NWH layout (fwd + bwd)cudnn.ops.b2b_causal_conv1d— Back-to-back fused (fwd + bwd)C++ samples (
samples/cpp/causal_conv1d/):causal_conv1d_nwh.cpp— NWH forward + backwardb2b_causal_conv1d.cpp— B2B forward + backwardJupyter notebooks (
samples/python/):60_causal_conv1d_forward.ipynb— NHW forward (refreshed outputs)61_causal_conv1d_backward.ipynb— NHW backward (refreshed outputs)62_causal_conv1d_nwh_forward.ipynb— NWH forward (new)63_causal_conv1d_nwh_backward.ipynb— NWH backward (new)64_b2b_causal_conv1d_forward.ipynb— B2B forward (new)65_b2b_causal_conv1d_backward.ipynb— B2B backward (new)Infrastructure:
cudnn_frontend_shim.h,pycudnn.cppupdates for new opsTesting
All 6 notebooks executed and passing on SM100 with cuDNN dev.
Related
Mirrors GitLab MR !2112. Depends on cuDNN backend MR !3206 (cuhyena kernel integration).