diff --git a/tests/py/dynamo/conversion/test_cumsum_aten.py b/tests/py/dynamo/conversion/test_cumsum_aten.py index 634a85af4e..1a83c9c45f 100644 --- a/tests/py/dynamo/conversion/test_cumsum_aten.py +++ b/tests/py/dynamo/conversion/test_cumsum_aten.py @@ -1,5 +1,3 @@ -import unittest - import torch import torch.nn as nn import torch_tensorrt @@ -9,10 +7,6 @@ from .harness import DispatchTestCase -@unittest.skipIf( - torch_tensorrt.ENABLED_FEATURES.tensorrt_rtx, - "cumsum is not supported on TensorRT-RTX (build_serialized_network returns None on Linux as well as Windows)", -) class TestCumsumConverter(DispatchTestCase): @parameterized.expand( [