From 8be31b4a37abffe826cf9061759163b1bfaf60ec Mon Sep 17 00:00:00 2001
From: Zhengxing Chen <czxttkl@fb.com>
Date: Tue, 26 Jul 2022 22:52:45 -0700
Subject: [PATCH] Torchrec gpu stable (#660)

Summary:
Pull Request resolved: https://github.com/facebookresearch/ReAgent/pull/660

use this

Differential Revision: D38185498

fbshipit-source-id: 72785c9d62ddcc14d2a8529268261137e74b30a2
---
 .circleci/config.yml | 4 ++--
 setup.cfg            | 8 ++++++--
 tox.ini              | 4 ++--
 3 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/.circleci/config.yml b/.circleci/config.yml
index 759335ed..14b07c31 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -172,13 +172,13 @@ commands:
                       # "-e" option will activate the development mode (a symlink to the code in ReAgent
                       # will be created in site-packages directory)
                       command: |
-                        pip install -e .[gym,test,torchrec_gpu] --pre --extra-index-url https://download.pytorch.org/whl/cu113 -f https://download.pytorch.org/whl/nightly/torchrec_nightly/ --progress-bar off
+                        pip install -e .[gym,test,torchrec_gpu] --pre --extra-index-url https://download.pytorch.org/whl/cu113 -f https://download.pytorch.org/whl/torchrec/ --progress-bar off
             - unless:
                 condition: << parameters.is_ubuntu_gpu >>
                 steps:
                   - run:
                       command: |
-                        pip install -e .[gym,test,torchrec_cpu] --pre --extra-index-url https://download.pytorch.org/whl/cpu -f https://download.pytorch.org/whl/nightly/torchrec_nightly_cpu/ --progress-bar off
+                        pip install -e .[gym,test,torchrec_cpu] --pre --extra-index-url https://download.pytorch.org/whl/cpu -f https://download.pytorch.org/whl/torchrec_cpu/ --progress-bar off
 
 
   run_unittest:
diff --git a/setup.cfg b/setup.cfg
index 84a1c702..26c38fe5 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -53,9 +53,13 @@ ax = ax-platform
 
 lite = nevergrad>=0.4.3
 
-torchrec_gpu = torchrec-nightly
+torchrec_gpu =
+  fbgemm-gpu
+  torchrec
 
-torchrec_cpu = torchrec-nightly-cpu
+torchrec_cpu =
+  fbgemm-gpu-cpu
+  torchrec-cpu
 
 
 ###########
diff --git a/tox.ini b/tox.ini
index 52add020..89862583 100644
--- a/tox.ini
+++ b/tox.ini
@@ -26,7 +26,7 @@ extras =
     test
     torchrec_cpu
 install_command =
-    pip install --pre --extra-index-url https://download.pytorch.org/whl/cpu -f https://download.pytorch.org/whl/nightly/torchrec_nightly_cpu/ {opts} {packages} --progress-bar off
+    pip install --pre --extra-index-url https://download.pytorch.org/whl/cpu -f https://download.pytorch.org/whl/torchrec_cpu/ {opts} {packages} --progress-bar off
 commands =
     pytest -n2 -m "(not serial) and (not seq2slate_long)"
     pytest -n0 -m "serial"
@@ -39,7 +39,7 @@ extras =
     test
     torchrec_gpu
 install_command =
-    pip install --pre --extra-index-url https://download.pytorch.org/whl/cu113 -f https://download.pytorch.org/whl/nightly/torchrec_nightly/ {opts} {packages} --progress-bar off
+    pip install --pre --extra-index-url https://download.pytorch.org/whl/cu113 -f https://download.pytorch.org/whl/torchrec/ {opts} {packages} --progress-bar off
 
 
 [testenv:circleci_misc_unittest]