Handle the device passed as string

Signed-off-by: Przemek Tredak <ptredak@nvidia.com>

Author: ptrendx

transformer_engine/pytorch/quantized_tensor.py

@@ -286,14 +286,16 @@ def make_empty(
         shape: Iterable[int],
         *,
         dtype: torch.dtype = torch.float32,
-        device: Optional[torch.device] = None,
+        device: Optional[Union[torch.device, str]] = None,
         requires_grad: bool = False,
         pin_memory: bool = False,
     ) -> QuantizedTensor:
         """Construct quantized tensor with uninitialized data"""
 
         if device is None:
             device = torch.device("cuda")
+        # Handle the device passed as string
+        device = torch.device(device)
         result = tex.create_empty_quantized_tensor(
             self,
             list(shape),