From c53265ff6a5572721253d82c8bc67bd277f9697d Mon Sep 17 00:00:00 2001 From: Victor Stone Date: Tue, 21 Jul 2026 12:24:34 -0700 Subject: [PATCH] test: re-enable cumsum tests on TensorRT-RTX --- tests/py/dynamo/conversion/test_cumsum_aten.py | 6 ------ 1 file changed, 6 deletions(-) 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( [