Add xpu modifications to fortran for example 2, init still not called

Cambridge-ICCS · Dec 20, 2024 · d08ce7e · d08ce7e
1 parent 34316e0
commit d08ce7e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/2_ResNet18/resnet_infer_fortran.f90 b/examples/2_ResNet18/resnet_infer_fortran.f90
@@ -3,7 +3,7 @@ program inference
    use, intrinsic :: iso_fortran_env, only : sp => real32
 
    ! Import our library for interfacing with PyTorch
-   use ftorch, only : torch_model, torch_tensor, torch_kCPU, torch_delete, &
+   use ftorch, only : torch_model, torch_tensor, torch_kXPU, torch_kCPU, torch_delete, &
                       torch_tensor_from_array, torch_model_load, torch_model_forward
 
    ! Import our tools module for testing utils
@@ -82,12 +82,12 @@ subroutine main()
       call load_data(filename, tensor_length, in_data)
 
       ! Create input/output tensors from the above arrays
-      call torch_tensor_from_array(in_tensors(1), in_data, in_layout, torch_kCPU)
+      call torch_tensor_from_array(in_tensors(1), in_data, in_layout, torch_kXPU, device_index=0)
 
       call torch_tensor_from_array(out_tensors(1), out_data, out_layout, torch_kCPU)
 
       ! Load ML model (edit this line to use different models)
-      call torch_model_load(model, args(1))
+      call torch_model_load(model, args(1), device_index=0)
 
       ! Infer
       call torch_model_forward(model, in_tensors, out_tensors)