From 6ff03e6d19e96425a26f9c0159ec9ff988fe64b6 Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Tue, 10 Sep 2024 14:03:33 -0700
Subject: [PATCH] Update torchxla2 installation for torch on mac and others
 (#7990)

---
 experimental/torch_xla2/dev-requirements.txt | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/experimental/torch_xla2/dev-requirements.txt b/experimental/torch_xla2/dev-requirements.txt
index 3d056e6640b..70dc73047d2 100644
--- a/experimental/torch_xla2/dev-requirements.txt
+++ b/experimental/torch_xla2/dev-requirements.txt
@@ -1,3 +1,4 @@
 -f https://download.pytorch.org/whl/torch
-torch==2.4.0+cpu
+torch==2.4.0; sys_platform == 'darwin'  # macOS
+torch==2.4.0+cpu; sys_platform != 'darwin' # Non-macOS (CPU-only), like on TPU
 ruff~=0.3.5