Fix IREE eager backend device string (huggingface#237)

Author: qedawkins

shark/iree_eager_backend.py

@@ -48,8 +48,8 @@ class EagerModeIREELinalgOnTensorsBackend(TorchMLIREagerBackend):
 
     def __init__(self, device: str):
         self.torch_device_str = device
-        self.iree_device_str = IREE_DEVICE_MAP[device]
-        self.config = ireert.Config(self.iree_device_str)
+        self.config = ireert.Config(IREE_DEVICE_MAP[device])
+        self.raw_device_str = device
 
     def get_torch_metadata(
         self, tensor: DeviceArray, kwargs: Dict[str, Any]
@@ -71,7 +71,7 @@ def compile(self, imported_module: Module):
             "EagerMode",
         )
         callable, _ = get_iree_compiled_module(
-            imported_module, self.iree_device_str, func_name=fn_name
+            imported_module, self.raw_device_str, func_name=fn_name
         )
         return callable