chore: add device type to resnet example

Cambridge-ICCS · Jan 24, 2025 · 3946312 · 3946312
1 parent 5c0f007
commit 3946312
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/2_ResNet18/resnet_infer_fortran.f90 b/examples/2_ResNet18/resnet_infer_fortran.f90
@@ -87,7 +87,7 @@ subroutine main()
       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), device_index=0)
+      call torch_model_load(model, args(1), device_type=torch_kXPU, device_index=0)
 
       ! Infer
       call torch_model_forward(model, in_tensors, out_tensors)