🐛 [Bug] Schema issue encountered when compiling Waveglow model (original) (raw)

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