fix check for disc backend

yitongh · yitongh · commit 46a33b1281c7 · 2024-08-20T15:21:51.000+08:00
diff --git a/test/test_utils.py b/test/test_utils.py
@@ -13,8 +13,7 @@
 import torch_xla.core.xla_model as xm
 import torch_xla.utils.utils as xu
 
-_IS_DISC_BACKEND = os.environ.get(
-    'DISC_DEVICE', '0').upper() in ['ON', '1', 'YES', 'TRUE', 'Y']
+_IS_DISC_BACKEND = 'DISC_DEVICE' in os.environ
 
 
 def is_disc_backend():
