Skip to content

🐛 [Bug] Schema issue encountered when compiling Waveglow model #1780

Closed
@gs-olive

Description

@gs-olive

Bug Description

When compiling the Waveglow model with Torch-TRT, the following error is encountered:

RuntimeError: 
    Schema not found for node. File a bug report.
    Node: %25634 : int = aten::div(%22926, %25633, %234)

    Input types: int, int, str

The missing TorchScript converter/evaluator is:
aten::div(int, int, "trunc")

To Reproduce

Steps to reproduce the behavior:

  1. Run torch_tensorrt.compile with Waveglow model as input, using fp32 precision.
  2. Choose fixed input sizes of (1, 80, 256) and (1, 8, 8192) and enable truncate_long_and_double with 12 GB workspace.

Expected behavior

Model should successfully compile to Torch-TRT. Specifically, this invalid aten::div schema should not arise in the TorchScript IR.

Environment

  • Torch-TensorRT Version (e.g. 1.0.0): 97209fe
  • PyTorch Version (e.g. 1.0): 2.0.0.dev20230219+cu117

Metadata

Metadata

Assignees

Labels

bugSomething isn't workingcomponent: evaluatorsIssues re: Specific op evaluators

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions