chore: resolve failures in CI - disable the plugin for Orin and for TRT-RTX

Author: narendasan

core/plugins/BUILD

@@ -62,6 +62,11 @@ config_setting(
 cc_library(
     name = "torch_tensorrt_plugins",
     srcs = select({
+        ":jetpack": [
+            "impl/interpolate_plugin.cpp",
+            "impl/normalize_plugin.cpp",
+            "register_plugins.cpp",
+        ],
         ":rtx_win": [],
         ":rtx_x86_64": [],
         "//conditions:default": [
@@ -72,6 +77,11 @@ cc_library(
         ],
     }),
     hdrs = select({
+        ":jetpack": [
+            "impl/interpolate_plugin.h",
+            "impl/normalize_plugin.h",
+            "plugins.h",
+        ],
         ":rtx_win": [],
         ":rtx_x86_64": [],
         "//conditions:default": [

core/plugins/register_plugins.cpp

@@ -1,8 +1,12 @@
 #include "NvInferPlugin.h"
 #include "NvInferPluginUtils.h"
+#include "NvInferVersion.h"
 #include "core/plugins/impl/interpolate_plugin.h"
 #include "core/plugins/impl/normalize_plugin.h"
+// ScatterAdd plugin is not available on Jetpack (TRT 10.3.x / L4T) or TRT-RTX builds
+#if NV_TENSORRT_MAJOR > 10 || (NV_TENSORRT_MAJOR == 10 && NV_TENSORRT_MINOR > 3)
 #include "core/plugins/impl/scatter_add_plugin.h"
+#endif
 #include "core/plugins/plugins.h"
 #include "core/util/prelude.h"
 

py/torch_tensorrt/_features.py

@@ -52,17 +52,20 @@
 _WINDOWS_CROSS_COMPILE = check_cross_compile_trt_win_lib()
 _TRTLLM_AVAIL = load_tensorrt_llm_for_nccl()
 
-if importlib.util.find_spec("tensorrt.plugin") and importlib.util.find_spec(
-    "tensorrt.plugin._lib"
-):
-    # there is a bug in tensorrt 10.14.* and 10.15.* that causes the plugin to not work, disable it for now
-    if tensorrt.__version__.startswith("10.15.") or tensorrt.__version__.startswith(
-        "10.14."
+try:
+    if importlib.util.find_spec("tensorrt.plugin") and importlib.util.find_spec(
+        "tensorrt.plugin._lib"
     ):
-        _QDP_PLUGIN_AVAIL = False
+        # there is a bug in tensorrt 10.14.* and 10.15.* that causes the plugin to not work, disable it for now
+        if tensorrt.__version__.startswith("10.15.") or tensorrt.__version__.startswith(
+            "10.14."
+        ):
+            _QDP_PLUGIN_AVAIL = False
+        else:
+            _QDP_PLUGIN_AVAIL = True
     else:
-        _QDP_PLUGIN_AVAIL = True
-else:
+        _QDP_PLUGIN_AVAIL = False
+except Exception:
     _QDP_PLUGIN_AVAIL = False
 
 ENABLED_FEATURES = FeatureSet(

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

@@ -1095,6 +1095,9 @@ def aten_ops_select(
 def _scatter_add_plugin_available() -> bool:
     import tensorrt as trt
 
+    # ScatterAdd plugin is not built for Jetpack (TRT 10.3.x) or TRT-RTX builds
+    if trt.__version__.startswith("10.3.") or trt._package_name == "tensorrt_rtx":
+        return False
     return (
         trt.get_plugin_registry().get_creator("ScatterAdd", "1", "torch_tensorrt")
         is not None

tests/py/dynamo/conversion/test_index_put_aten.py

@@ -1,7 +1,9 @@
+import pytest
 import torch
 import torch_tensorrt as torchtrt
 from parameterized import param, parameterized
 from torch.testing._internal.common_utils import run_tests
+from torch_tensorrt import ENABLED_FEATURES
 
 from .harness import DispatchTestCase
 
@@ -409,6 +411,9 @@ class TestIndexPutConverter(DispatchTestCase):
     def test_index_put(
         self, test_name, source_tensor, indices_tensor, value_tensor, accumulate=False
     ):
+        if accumulate and ENABLED_FEATURES.tensorrt_rtx:
+            pytest.skip("ScatterAdd plugin not available in TRT RTX")
+
         @torch._dynamo.assume_constant_result
         def get_indices_tensor():
             return indices_tensor
@@ -425,6 +430,7 @@ def forward(self, source_tensor, value_tensor):
             inputs=[source_tensor, value_tensor],
             enable_passes=True,
             use_dynamo_tracer=True,
+            use_explicit_typing=True,
         )
 
     def test_index_add_dynamic_shape(self):
@@ -868,6 +874,9 @@ def forward(self, src, values, idx0, idx1):
     # duplicate positions when index_put is embedded in a larger graph.
     # ------------------------------------------------------------------
 
+    @pytest.mark.skipif(
+        ENABLED_FEATURES.tensorrt_rtx, reason="ScatterAdd plugin not available in TRT RTX"
+    )
     def test_kv_cache_duplicate_slot_writes(self):
         """KV-cache style: linear projection → index_put(accumulate=True) into
         a flat cache with duplicate slot indices → output projection.
@@ -903,8 +912,12 @@ def forward(self, tokens, cache):
             inputs=[tokens, cache],
             use_dynamo_tracer=True,
             enable_passes=True,
+            use_explicit_typing=True,
         )
 
+    @pytest.mark.skipif(
+        ENABLED_FEATURES.tensorrt_rtx, reason="ScatterAdd plugin not available in TRT RTX"
+    )
     def test_sparse_embedding_duplicate_seq_ids(self):
         """Sparse embedding accumulation: embedding lookup → index_put(accumulate=True)
         into per-sequence accumulators where many tokens map to the same sequence → ReLU.
@@ -943,8 +956,12 @@ def forward(self, token_ids, accum):
             inputs=[token_ids, accum],
             use_dynamo_tracer=True,
             enable_passes=True,
+            use_explicit_typing=True,
         )
 
+    @pytest.mark.skipif(
+        ENABLED_FEATURES.tensorrt_rtx, reason="ScatterAdd plugin not available in TRT RTX"
+    )
     def test_histogram_conv_duplicate_bin_ids(self):
         """Histogram accumulation: Conv1d → index_put(accumulate=True) into histogram
         bins where many frames land in the same bin → mean pool → linear.
@@ -983,6 +1000,7 @@ def forward(self, signal, hist):
             inputs=[signal, hist],
             use_dynamo_tracer=True,
             enable_passes=True,
+            use_explicit_typing=True,
         )