[TRTLLM-11353][feat] API to configure TeaCache coefficients

[o-stoner]

@coderabbitai summary

Description

Allow users to supply TeaCache coefficients to 1) use TeaCache for currently unsupported models (Wan 2.2, LTX-2) or 2) override the coefficients of currently supported TeaCache models.

Test Coverage

PR Checklist

Please review the following before submitting your PR:

• PR description clearly explains what and why. If using CodeRabbit's summary, please make sure it makes sense.

• PR Follows TRT-LLM CODING GUIDELINES to the best of your knowledge.

• Test cases are provided for new code paths (see test instructions)

• Any new dependencies have been scanned for license and vulnerabilities

• CODEOWNERS updated if ownership changes

• Documentation updated as needed

• Update tava architecture diagram if there is a significant design change in PR.

• The reviewers assigned automatically/manually are appropriate for the PR.

• Please check this after reviewing the above items as appropriate for this PR.

GitHub Bot Help

To see a list of available CI bot commands, please comment /bot help.

[o-stoner]

/bot run --disable-fail-fast

[tensorrt-cicd]

PR_Github #44077 [ run ] triggered by Bot. Commit: 55cab64 Link to invocation

[tensorrt-cicd]

PR_Github #44077 [ run ] completed with state SUCCESS. Commit: 55cab64
/LLM/main/L0_MergeRequest_PR pipeline #34507 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

• Please check the failed tests and fix your PR
• If you cannot view the failures, ask the CI triggerer to share details
• Once fixed, request an NVIDIA team member to trigger CI again

Link to invocation

[o-stoner]

/bot run --disable-fail-fast

[tensorrt-cicd]

PR_Github #44510 [ run ] triggered by Bot. Commit: 1d118f8 Link to invocation

[tensorrt-cicd]

PR_Github #44510 [ run ] completed with state SUCCESS. Commit: 1d118f8
/LLM/main/L0_MergeRequest_PR pipeline #34909 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

• Please check the failed tests and fix your PR
• If you cannot view the failures, ask the CI triggerer to share details
• Once fixed, request an NVIDIA team member to trigger CI again

Link to invocation

[o-stoner]

/bot run --disable-fail-fast

[tensorrt-cicd]

PR_Github #44802 [ run ] triggered by Bot. Commit: 553618c Link to invocation

[tensorrt-cicd]

PR_Github #44802 [ run ] completed with state SUCCESS. Commit: 553618c
/LLM/main/L0_MergeRequest_PR pipeline #35152 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

• Please check the failed tests and fix your PR
• If you cannot view the failures, ask the CI triggerer to share details
• Once fixed, request an NVIDIA team member to trigger CI again

Link to invocation

[o-stoner]

/bot run --disable-fail-fast

[tensorrt-cicd]

PR_Github #45000 [ run ] triggered by Bot. Commit: 553618c Link to invocation

[tensorrt-cicd]

PR_Github #45000 [ run ] completed with state SUCCESS. Commit: 553618c
/LLM/main/L0_MergeRequest_PR pipeline #35319 completed with status: 'SUCCESS'

CI Report

Link to invocation

[o-stoner]

/bot run --disable-fail-fast

[tensorrt-cicd]

PR_Github #45462 [ run ] triggered by Bot. Commit: 68e8c7c Link to invocation

[tensorrt-cicd]

PR_Github #45462 [ run ] completed with state SUCCESS. Commit: 68e8c7c
/LLM/main/L0_MergeRequest_PR pipeline #35695 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

• Please check the failed tests and fix your PR
• If you cannot view the failures, ask the CI triggerer to share details
• Once fixed, request an NVIDIA team member to trigger CI again

Link to invocation