TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

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

Open peri044 opened this issue 1 year ago • 0 comments

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

peri044 avatar Jun 06 '24 04:06 peri044