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.

Sign up for GitHub

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

Jump to bottom

🐛 [Bug] Error: "Conversion of function torch._ops.aten.aten::cumsum not currently supported!" on CUDA 11.8 and 12.1 #3214

Open
zewenli98 opened this issue Oct 6, 2024 · 0 comments
Open

🐛 [Bug] Error: "Conversion of function torch._ops.aten.aten::cumsum not currently supported!" on CUDA 11.8 and 12.1 #3214

zewenli98 opened this issue Oct 6, 2024 · 0 comments
Assignees
@zewenli98
Labels
bug Something isn't working

Comments

@zewenli98
Copy link
Collaborator

zewenli98 commented Oct 6, 2024 •
edited
Loading

Bug Description

In the CI tests, all tests of cumsum was failed on CUDA 11.8 and 12.1, but works on 12.4. The error is like:

FAILED conversion/test_cumsum_aten.py::TestCumsumConverter::test_cumsum_1D_0 - torch_tensorrt.dynamo.conversion._TRTInterpreter.UnsupportedOperatorException: Conversion of function torch._ops.aten.aten::cumsum not currently supported!
FAILED conversion/test_cumsum_aten.py::TestCumsumConverter::test_cumsum_1D_1 - torch_tensorrt.dynamo.conversion._TRTInterpreter.UnsupportedOperatorException: Conversion of function torch._ops.aten.aten::cumsum not currently supported!
FAILED conversion/test_cumsum_aten.py::TestCumsumConverter::test_cumsum_1D_2 - torch_tensorrt.dynamo.conversion._TRTInterpreter.UnsupportedOperatorException: Conversion of function torch._ops.aten.aten::cumsum not currently supported!
...

It also works on my local machine with RTX 4080 + CUDA 12.2.
@zewenli98 zewenli98 added the bug Something isn't working label Oct 6, 2024
@zewenli98 zewenli98 self-assigned this Nov 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@zewenli98 zewenli98

Labels
bug Something isn't working
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@zewenli98