From 7b49903df663fe94bff734435689ea3e67565939 Mon Sep 17 00:00:00 2001 From: Wanchao Date: Mon, 24 Oct 2022 09:12:52 -0700 Subject: [PATCH] [spmd] pin pt version (#572) A recent change of c10d.ReduceOp crashes deepcopy of it, pin pt version temporarily to fix CI. see https://github.com/pytorch/pytorch/pull/87303#discussion_r1002879700 --- spmd/requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spmd/requirements.txt b/spmd/requirements.txt index 912dc9e6b..49f2202e9 100644 --- a/spmd/requirements.txt +++ b/spmd/requirements.txt @@ -5,10 +5,10 @@ pyyaml --pre --extra-index-url https://download.pytorch.org/whl/nightly/cpu -torch >= 1.14.0.dev0; sys_platform == "darwin" +torch >= 1.14.0.dev0, <= 1.14.0dev20221021; sys_platform == "darwin" --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu113 --extra-index-url https://download.pytorch.org/whl/nightly/cu116 -torch >= 1.14.0.dev0; sys_platform == "linux" +torch >= 1.14.0.dev0, <= 1.14.0dev20221021; sys_platform == "linux"