address reviewer's comments

Author: chilo-ms

plugin_execution_providers/tensorrt/CMakeLists.txt

@@ -31,7 +31,14 @@ add_definitions(-DONNX_NAMESPACE=onnx)
 add_definitions(-DONNX_ML)
 add_definitions(-DNOMINMAX)
 
-file(GLOB tensorrt_src "./src/*.cc" "./src/kernels/*.cc" "./src/utils/*.cc" "./src/cuda/unary_elementwise_ops_impl.cu" "./src/*.h" "./src/kernels/*.h")
+file(GLOB tensorrt_src 
+  "./src/*.cc"
+  "./src/kernels/*.cc"
+  "./src/utils/*.cc"
+  "./src/cuda/unary_elementwise_ops_impl.cu"
+  "./src/*.h"
+  "./src/kernels/*.h"
+)
 add_library(TensorRTEp SHARED ${tensorrt_src})
 
 set_onnxruntime_paths(

plugin_execution_providers/tensorrt/src/tensorrt_execution_provider_data_transfer.cc

@@ -24,8 +24,11 @@ bool ORT_API_CALL TRTEpDataTransfer::CanCopyImpl(const OrtDataTransferImpl* this
   auto dst_vendor_id = impl.ep_api.MemoryDevice_GetVendorId(dst_memory_device);
 
   // 0x10DE is the PCI vendor ID for NVIDIA
-  if ((src_type == OrtMemoryInfoDeviceType_GPU && src_vendor_id != 0x10DE) ||
-      (dst_type == OrtMemoryInfoDeviceType_GPU && dst_vendor_id != 0x10DE)) {
+  constexpr uint32_t nvidia_vendor_id = 0x10DE;
+
+  // Reject if GPU device is not NVIDIA
+  if ((src_type == OrtMemoryInfoDeviceType_GPU && src_vendor_id != nvidia_vendor_id) ||
+      (dst_type == OrtMemoryInfoDeviceType_GPU && dst_vendor_id != nvidia_vendor_id)) {
     return false;
   }
 

plugin_execution_providers/tensorrt/src/tensorrt_execution_provider_kernel_registration.h

@@ -3,8 +3,6 @@
 
 #pragma once
 
-#include <vector>
-
 #include "ep_utils.h"
 
 namespace trt_ep {