-
Notifications
You must be signed in to change notification settings - Fork 390
🐛 [Bug] Schema issue encountered when compiling Waveglow model #1780
Copy link
Copy link
Closed
Labels
bugSomething isn't workingSomething isn't workingcomponent: evaluatorsIssues re: Specific op evaluatorsIssues re: Specific op evaluators
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, strThe missing TorchScript converter/evaluator is:
aten::div(int, int, "trunc")
To Reproduce
Steps to reproduce the behavior:
- Run
torch_tensorrt.compilewith Waveglow model as input, using fp32 precision. - Choose fixed input sizes of
(1, 80, 256)and(1, 8, 8192)and enabletruncate_long_and_doublewith 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
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't workingcomponent: evaluatorsIssues re: Specific op evaluatorsIssues re: Specific op evaluators