Skip to content

Add cuDNN tensor IR support - #15

Open
cerisier wants to merge 1 commit into
mainfrom
cerisier/cudnn-tensor-ir
Open

cerisier wants to merge 1 commit into
mainfrom
cerisier/cudnn-tensor-ir

Conversation

@cerisier

Copy link
Copy Markdown
Member

Summary

  • Expose cudnn_engines_tensor_ir and cudnn_engines_tensor_ir_static through cuDNN 9 BUILD/proxy targets.
  • Gate tensor IR on cuDNN >= 9.21.0 and actual archive contents, so cuDNN 9.20 redists keep building without the new library.
  • Add e2e coverage at the introduction boundary by pinning the CUDA 13.3 smoke case to cuDNN 9.21.0 and checking the shared/static tensor IR labels.

Context

This mirrors the tensor_ir support being added in rules_ml_toolchain, but keeps the target conditional because libcudnn_engines_tensor_ir first appears in cuDNN 9.21.

Checked locally with bazel build //..., plus CUDA 13.3/cuDNN 9.21 tensor IR and CUDA 13.2/cuDNN 9.20 cuDNN smoke builds.

@cerisier
cerisier marked this pull request as ready for review June 26, 2026 02:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant