Update ftorch.F90 for XPU support

Cambridge-ICCS · Dec 20, 2024 · 4c2fc90 · 4c2fc90
1 parent 84a4e5d
commit 4c2fc90
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/ftorch.F90 b/src/ftorch.F90
@@ -49,6 +49,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