libinephany 0.15.3__tar.gz → 0.15.4__tar.gz

libinephany-0.15.4/CODE_VERSION.cfg

@@ -0,0 +1 @@
+0.15.4

{libinephany-0.15.3/libinephany.egg-info → libinephany-0.15.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: libinephany
-Version: 0.15.3
+Version: 0.15.4
 Summary: Inephany library containing code commonly used by multiple subpackages.
 Author-email: Inephany <info@inephany.com>
 License: Apache 2.0

{libinephany-0.15.3 → libinephany-0.15.4}/libinephany/observations/observation_utils.py

@@ -167,6 +167,20 @@ def create_one_hot_depth_encoding(agent_controlled_modules: list[str], parameter
     return create_one_hot_observation(vector_length=3, one_hot_index=one_hot_index)
 
 
+def check_if_tensor_is_xpu(tensor: torch.Tensor | None) -> bool:
+    """
+    :param tensor: Tensor to check whether it is an XPU tensor.
+    :return: Whether the tensor is an XPU tensor.
+    """
+
+    try:
+        xpu_available = torch.xpu.is_available()
+        return xpu_available and tensor.is_xpu
+
+    except AttributeError:
+        return False
+
+
 def tensor_on_local_rank(tensor: torch.Tensor | None) -> bool:
     """
     :param tensor: Tensor to check whether it is owned by the local rank, partially or entirely.
@@ -176,10 +190,10 @@ def tensor_on_local_rank(tensor: torch.Tensor | None) -> bool:
     valid_tensor = tensor is not None and tensor.numel() > 0
     device_index = tensor.device.index if valid_tensor else None
 
-    xpu_available = torch.xpu.is_available()
+    is_xpu = check_if_tensor_is_xpu(tensor)
     cuda_available = torch.cuda.is_available()
 
-    if valid_tensor and xpu_available and tensor.is_xpu:
+    if valid_tensor and is_xpu:
         current_device = torch.xpu.current_device()
         return device_index == current_device
 

{libinephany-0.15.3 → libinephany-0.15.4/libinephany.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: libinephany
-Version: 0.15.3
+Version: 0.15.4
 Summary: Inephany library containing code commonly used by multiple subpackages.
 Author-email: Inephany <info@inephany.com>
 License: Apache 2.0

libinephany-0.15.3/CODE_VERSION.cfg

@@ -1 +0,0 @@
-0.15.3