Update recipes_source/recipes/profiler_recipe.py

Co-authored-by: Yu, Guangye <[email protected]>

Author: louie-tsai

recipes_source/recipes/profiler_recipe.py

@@ -163,7 +163,12 @@
 ######################################################################
 # Profiler can also be used to analyze performance of models executed on GPUs and XPUs:
 # Users could switch between cpu, cuda and xpu
-device = 'cuda'
+if torch.cuda.is_available():
+    device = 'cuda'
+elif torch.xpu.is_available():
+    device = 'xpu'
+else:
+    device = 'cpu'
 
 activities = [ProfilerActivity.CPU]
 sort_by_keyword = device + "_time_total"