Update device logging and defaults

Author: gs-olive

py/torch_tensorrt/dynamo/_defaults.py

@@ -12,4 +12,4 @@
 TRUNCATE_LONG_AND_DOUBLE = False
 USE_PYTHON_RUNTIME = False
 USE_FAST_PARTITIONER = True
-DEVICE = Device._current_device()
+DEVICE = Device(gpu_id=torch.cuda.current_device())

py/torch_tensorrt/dynamo/utils.py

@@ -182,14 +182,11 @@ def parse_dynamo_kwargs(kwargs: Any) -> CompilationSettings:
     settings.device = to_torch_tensorrt_device(settings.device)
 
     # Check and update device settings
-    default_torch_gpu_idx = torch.cuda.default_stream().device.index
-    if "device" not in kwargs and default_torch_gpu_idx != settings.device.gpu_id:
-        logger.warning(
-            f"No device specified, detected differing gpu IDs for CUDA default: {settings.device.gpu_id} "
-            f"and Torch default: {default_torch_gpu_idx}. Using Torch default gpu ID: {default_torch_gpu_idx}. "
+    if "device" not in kwargs:
+        logger.info(
+            f"Device not specified, using Torch default current device - cuda:{settings.device.gpu_id}. "
             "If this is incorrect, please specify an input device, via the device keyword."
         )
-        settings.device = Device(gpu_id=default_torch_gpu_idx)
 
     logger.debug(f"Compiling with Settings:\n{settings}")