[CI] Bump test containers to latest

Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>

Author: kevalmorabia97

.github/workflows/example_tests.yml

@@ -60,7 +60,7 @@ jobs:
     uses: ./.github/workflows/_example_tests_runner.yml
     secrets: inherit
     with:
-      docker_image: "nvcr.io/nvidia/tensorrt-llm/release:1.3.0rc10"
+      docker_image: "nvcr.io/nvidia/tensorrt-llm/release:1.3.0rc12"
       example: ${{ matrix.example }}
       pip_install_extras: "[hf,dev-test]"
       runner: linux-amd64-gpu-rtxpro6000-latest-1
@@ -74,7 +74,7 @@ jobs:
     uses: ./.github/workflows/_example_tests_runner.yml
     secrets: inherit
     with:
-      docker_image: "nvcr.io/nvidia/tensorrt-llm/release:1.3.0rc10"
+      docker_image: "nvcr.io/nvidia/tensorrt-llm/release:1.3.0rc12"
       example: ${{ matrix.example }}
       pip_install_extras: "[hf,dev-test]"
       runner: linux-amd64-gpu-rtxpro6000-latest-2

.github/workflows/gpu_tests.yml

@@ -40,14 +40,13 @@ jobs:
         include:
           - example: gpu
             timeout: 60
-            container_image: pytorch:26.01-py3
-            # tests/gpu/_extensions/test_onnx_extensions.py fails for newer containers until https://github.com/tbenthompson/cppimport/pull/98
+            container_image: pytorch:26.03-py3
           - example: gpu_megatron
             timeout: 45
             container_image: nemo:26.04
           - example: gpu_trtllm
             timeout: 30
-            container_image: tensorrt-llm/release:1.3.0rc10
+            container_image: tensorrt-llm/release:1.3.0rc12
     runs-on: ${{ startsWith(github.ref, 'refs/heads/pull-request/') && 'linux-amd64-gpu-rtxpro6000-latest-1' || 'linux-amd64-gpu-rtxpro6000-latest-2' }}
     timeout-minutes: ${{ matrix.timeout }}
     container:

modelopt/onnx/quantization/extensions.py

@@ -18,7 +18,7 @@
 import os
 import sys
 
-# TODO: cppimport is no longer maintained, switch to a different library
+# TODO: cppimport is not actively maintained, consider a better alternative library
 import cppimport
 
 from modelopt.onnx.logging_config import logger
@@ -32,7 +32,5 @@
 except Exception as e:
     logger.warning(
         f"{e}\nUnable to load `modelopt_round_and_pack_ext', falling back to python based optimized version. "
-        "If you see `copy_file() got an unexpected keyword argument 'dry_run'`, you will need "
-        "https://github.com/tbenthompson/cppimport/pull/98 or downgrade setuptools until we have a workaround"
     )
     round_and_pack_ext = None