Skip to content

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

Open
@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

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions