Closed
Description
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:
- Run
torch_tensorrt.compile
with Waveglow model as input, using fp32 precision. - Choose fixed input sizes of
(1, 80, 256)
and(1, 8, 8192)
and enabletruncate_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