Skip to content

🐛 [Bug] BERT failure with cuda 12.4 (Schema not found for node add) #2893

@peri044

Description

@peri044

Bug Description

Using nightly, the CI produces this issue with cuda 12.4

=========================== short test summary info ============================
FAILED models/test_models.py::TestModels::test_bert_base_uncased - RuntimeError: 
Schema not found for node. File a bug report.
Node: %6932 : int = aten::add(%5840, %6931, %26)

To Reproduce

Steps to reproduce the behavior:

Expected behavior

Environment

Build information about Torch-TensorRT can be found by turning on debug messages

  • Torch-TensorRT Version (e.g. 1.0.0):
  • PyTorch Version (e.g. 1.0):
  • CPU Architecture:
  • OS (e.g., Linux):
  • How you installed PyTorch (conda, pip, libtorch, source):
  • Build command you used (if compiling from source):
  • Are you using local sources or building from archives:
  • Python version:
  • CUDA version:
  • GPU models and configuration:
  • Any other relevant information:

Additional context

Metadata

Metadata

Assignees

No one assigned

    Labels

    Story: Build & Install & PackagingInstallation failures, build system issues, Jetson/aarch64 support, Windows packaging, and dependenbugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions