Add MPS and XPU devices

src/ctorch.cpp

@@ -52,6 +52,18 @@ const auto get_device(torch_device_t device_type, int device_index)
                 << std::endl;
       exit(EXIT_FAILURE);
     }
+  case torch_kMPS:
+    if (device_index != -1 && device_index != 0) {
+      std::cerr << "[WARNING]: Only one device is available for MPS runs"
+                << std::endl;
+    }
+    return torch::Device(torch::kMPS);
+  case torch_kXPU:
+    if (device_index != -1) {
+      std::cerr << "[WARNING]: device index unused for XPU runs"
+                << std::endl;
+    }
+    return torch::Device(torch::kXPU);
   default:
     std::cerr << "[WARNING]: unknown device type, setting to torch_kCPU"
               << std::endl;

src/ctorch.h

@@ -26,7 +26,7 @@ typedef enum {
 } torch_data_t;
 
 // Device types
-typedef enum { torch_kCPU, torch_kCUDA } torch_device_t;
+typedef enum { torch_kCPU, torch_kCUDA, torch_kMPS, torch_kXPU } torch_device_t;
 
 
 // =====================================================================================

src/ftorch.f90

@@ -44,6 +44,8 @@ module ftorch
   enum, bind(c)
     enumerator :: torch_kCPU = 0
     enumerator :: torch_kCUDA = 1
+    enumerator :: torch_kMPS = 2
+    enumerator :: torch_kXPU = 3
   end enum
 
   !> Interface for directing `torch_tensor_from_array` to possible input types and ranks

src/ftorch.fypp

@@ -61,6 +61,8 @@ module ftorch
   enum, bind(c)
     enumerator :: torch_kCPU = 0
     enumerator :: torch_kCUDA = 1
+    enumerator :: torch_kMPS = 2
+    enumerator :: torch_kXPU = 3
   end enum
 
   !> Interface for directing `torch_tensor_from_array` to possible input types and ranks