update the order of xpu condition checking (#3220)

Co-authored-by: Ankith Gunapal <agunapal@ischool.Berkeley.edu>
pytorch · Jul 2, 2024 · f80e906 · f80e906
1 parent e979f41
commit f80e906
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ts/torch_handler/base_handler.py b/ts/torch_handler/base_handler.py
@@ -158,9 +158,9 @@ def initialize(self, context):
                 self.map_location + ":" + str(properties.get("gpu_id"))
             )
         elif (
-            torch.xpu.is_available()
+            os.environ.get("TS_IPEX_GPU_ENABLE", "false") == "true"
             and properties.get("gpu_id") is not None
-            and os.environ.get("TS_IPEX_GPU_ENABLE", "false") == "true"
+            and torch.xpu.is_available()
         ):
             self.map_location = "xpu"
             self.device = torch.device(