Update test/test_xpu.py

pkourdis · guangyey · pytorchmergebot · commit 0f9efdc8a2ef · 2025-05-17T04:12:55.000Z
Co-authored-by: Yu, Guangye &lt;106960996+guangyey@users.noreply.github.com&gt;
diff --git a/test/test_xpu.py b/test/test_xpu.py
@@ -32,7 +32,7 @@
 class TestXpuConfig(TestCase):
     def test_xpu_record(self):
         config = torch.__config__.show()
-        assert 'USE_XPU=0' in config or 'USE_XPU=1' in config
+        assert f'USE_XPU={int(torch.xpu._is_compiled())}` in config
 
     def test_xccl_record(self):
         config = torch.__config__.show()
