Narendasan/push ltlwrszpzrym

tests/py/dynamo/conversion/test_cumsum_aten.py

@@ -1,4 +1,3 @@
-import sys
 import unittest
 
 import torch
@@ -11,8 +10,8 @@
 
 
 @unittest.skipIf(
-    torch_tensorrt.ENABLED_FEATURES.tensorrt_rtx and sys.platform == "win32",
-    "cumsum errors out on TensorRT-RTX on Windows",
+    torch_tensorrt.ENABLED_FEATURES.tensorrt_rtx,
+    "cumsum errors out on TensorRT-RTX (build_serialized_network returns None)",
 )
 class TestCumsumConverter(DispatchTestCase):
     @parameterized.expand(

tests/py/dynamo/models/test_model_refit.py

@@ -1,6 +1,5 @@
 import importlib
 import os
-import sys
 import unittest
 
 import pytest
@@ -1014,8 +1013,8 @@ def forward(self, x):
     "Refit feature is not supported in Python 3.13 or higher",
 )
 @unittest.skipIf(
-    torch_trt.ENABLED_FEATURES.tensorrt_rtx and sys.platform == "win32",
-    "cumsum refit errors out on TensorRT-RTX on Windows",
+    torch_trt.ENABLED_FEATURES.tensorrt_rtx,
+    "cumsum refit errors out on TensorRT-RTX (build_serialized_network returns None)",
 )
 @pytest.mark.unit
 def test_refit_cumsum():

tests/py/dynamo/models/test_runtime_cache_models.py

@@ -14,8 +14,8 @@
 
 
 @unittest.skipIf(
-    not ENABLED_FEATURES.tensorrt_rtx,
-    "Runtime cache is only available with TensorRT-RTX",
+    not ENABLED_FEATURES.tensorrt_rtx or ENABLED_FEATURES.torch_tensorrt_runtime,
+    "Runtime cache uses the Python TRTEngine, only exercised in RTX python-only builds",
 )
 @unittest.skipIf(
     not importlib.util.find_spec("torchvision"),
@@ -136,8 +136,8 @@ def test_mobilenet_v2_with_runtime_cache(self):
 
 
 @unittest.skipIf(
-    not ENABLED_FEATURES.tensorrt_rtx,
-    "Runtime cache is only available with TensorRT-RTX",
+    not ENABLED_FEATURES.tensorrt_rtx or ENABLED_FEATURES.torch_tensorrt_runtime,
+    "Runtime cache uses the Python TRTEngine, only exercised in RTX python-only builds",
 )
 class TestRuntimeCacheDynamicShapes(TestCase):
     """Tests runtime cache with dynamic input shapes."""
@@ -254,8 +254,8 @@ def forward(self, x):
 
 
 @unittest.skipIf(
-    not ENABLED_FEATURES.tensorrt_rtx,
-    "Runtime cache is only available with TensorRT-RTX",
+    not ENABLED_FEATURES.tensorrt_rtx or ENABLED_FEATURES.torch_tensorrt_runtime,
+    "Runtime cache uses the Python TRTEngine, only exercised in RTX python-only builds",
 )
 class TestRuntimeCachePerformance(TestCase):
     """Informational timing tests for runtime cache warm-up behavior."""

tests/py/dynamo/runtime/test_000_runtime_cache.py

@@ -62,8 +62,8 @@ def _find_python_trt_engine(compiled):
 
 
 @unittest.skipIf(
-    not ENABLED_FEATURES.tensorrt_rtx,
-    "Runtime cache is only available with TensorRT-RTX",
+    not ENABLED_FEATURES.tensorrt_rtx or ENABLED_FEATURES.torch_tensorrt_runtime,
+    "Runtime cache uses the Python TRTEngine, only exercised in RTX python-only builds",
 )
 class TestRuntimeCacheSetup(TestCase):
     """Tests that runtime config and cache are correctly created for RTX."""
@@ -104,8 +104,8 @@ def test_runtime_cache_path_custom(self):
 
 
 @unittest.skipIf(
-    not ENABLED_FEATURES.tensorrt_rtx,
-    "Runtime cache is only available with TensorRT-RTX",
+    not ENABLED_FEATURES.tensorrt_rtx or ENABLED_FEATURES.torch_tensorrt_runtime,
+    "Runtime cache uses the Python TRTEngine, only exercised in RTX python-only builds",
 )
 class TestRuntimeCachePersistence(TestCase):
     """Tests that runtime cache is correctly saved to and loaded from disk."""
@@ -177,8 +177,8 @@ def test_save_creates_directory(self):
 
 
 @unittest.skipIf(
-    not ENABLED_FEATURES.tensorrt_rtx,
-    "Runtime cache is only available with TensorRT-RTX",
+    not ENABLED_FEATURES.tensorrt_rtx or ENABLED_FEATURES.torch_tensorrt_runtime,
+    "Runtime cache uses the Python TRTEngine, only exercised in RTX python-only builds",
 )
 class TestRuntimeCacheConcurrency(TestCase):
     """Tests that file locking works for concurrent access."""

tests/py/dynamo/runtime/test_001_cuda_graph_strategy.py

@@ -52,7 +52,7 @@ def _find_python_trt_engine(compiled):
 
 
 @unittest.skipIf(
-    not ENABLED_FEATURES.tensorrt_rtx,
+    not ENABLED_FEATURES.tensorrt_rtx or ENABLED_FEATURES.torch_tensorrt_runtime,
     "CUDA graph strategy requires TensorRT-RTX",
 )
 class TestCudaGraphStrategySetup(TestCase):
@@ -119,7 +119,7 @@ def test_default_compilation_settings(self):
 
 
 @unittest.skipIf(
-    not ENABLED_FEATURES.tensorrt_rtx,
+    not ENABLED_FEATURES.tensorrt_rtx or ENABLED_FEATURES.torch_tensorrt_runtime,
     "CUDA graph strategy integration requires TensorRT-RTX",
 )
 class TestCudaGraphStrategyWithSubgraphCudagraphs(TestCase):
@@ -176,7 +176,7 @@ def test_subgraph_mode_always_uses_rtx_native(self):
 
 
 @unittest.skipIf(
-    not ENABLED_FEATURES.tensorrt_rtx,
+    not ENABLED_FEATURES.tensorrt_rtx or ENABLED_FEATURES.torch_tensorrt_runtime,
     "Monolithic capturability tests require TensorRT-RTX",
 )
 class TestMonolithicCapturability(TestCase):
@@ -221,7 +221,7 @@ def test_none_strategy_monolithic_capturable(self):
 
 
 @unittest.skipIf(
-    not ENABLED_FEATURES.tensorrt_rtx,
+    not ENABLED_FEATURES.tensorrt_rtx or ENABLED_FEATURES.torch_tensorrt_runtime,
     "Context recreation tests require TensorRT-RTX",
 )
 class TestContextRecreation(TestCase):
@@ -273,7 +273,7 @@ def test_explicit_whole_graph_capture_no_override_needed(self):
 
 
 @unittest.skipIf(
-    not ENABLED_FEATURES.tensorrt_rtx,
+    not ENABLED_FEATURES.tensorrt_rtx or ENABLED_FEATURES.torch_tensorrt_runtime,
     "Cudagraph mode toggle tests require TensorRT-RTX",
 )
 class TestCudagraphModeToggle(TestCase):

tests/py/dynamo/runtime/test_001_dynamic_shapes_kernel_strategy.py

@@ -51,7 +51,7 @@ def _find_python_trt_engine(compiled):
 
 
 @unittest.skipIf(
-    not ENABLED_FEATURES.tensorrt_rtx,
+    not ENABLED_FEATURES.tensorrt_rtx or ENABLED_FEATURES.torch_tensorrt_runtime,
     "Dynamic shapes kernel specialization strategy requires TensorRT-RTX",
 )
 class TestDynamicShapesKernelStrategySetup(TestCase):