Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Closed
gs-olive opened this issue Mar 27, 2023 · 0 comments · Fixed by #1786
Closed

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

gs-olive opened this issue Mar 27, 2023 · 0 comments · Fixed by #1786
Assignees
Labels
bug Something isn't working component: evaluators Issues re: Specific op evaluators

Comments

@gs-olive
Copy link
Collaborator

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
@gs-olive gs-olive added bug Something isn't working component: evaluators Issues re: Specific op evaluators labels Mar 27, 2023
@gs-olive gs-olive assigned gs-olive and unassigned peri044 Mar 27, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working component: evaluators Issues re: Specific op evaluators
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants