refine _DEVICES

Author: zxd1997066

test/prototype/test_quantized_training.py

@@ -39,12 +39,8 @@
 if common_utils.SEED is None:
     common_utils.SEED = 1234
 
-_DEVICES = (
-    ["cpu"]
-    + (["cuda"] if torch.cuda.is_available() else [])
-    + (["xpu"] if torch.xpu.is_available() else [])
-)
 _DEVICE = get_current_accelerator_device()
+_DEVICES = ["cpu"] + ([_DEVICE] if torch.accelerator.is_available() else [])
 
 
 def _reset():