[Feature] Support for in-place functionalization (#714)

pytorch · Nov 30, 2022 · dc6b736 · dc6b736
1 parent 2788f06
commit dc6b736
Show file tree

Hide file tree

Showing 30 changed files with 1,328 additions and 2,063 deletions.
diff --git a/.circleci/unittest/linux_libs/scripts_gym/install.sh b/.circleci/unittest/linux_libs/scripts_gym/install.sh
@@ -44,5 +44,9 @@ fi
 # install tensordict
 pip install git+https://github.com/pytorch-labs/tensordict
 
+# smoke test
+python -c "import tensordict"
+
 printf "* Installing torchrl\n"
 python setup.py develop
+python -c "import torchrl"
diff --git a/.circleci/unittest/linux_libs/scripts_gym/run_test.sh b/.circleci/unittest/linux_libs/scripts_gym/run_test.sh
@@ -5,6 +5,8 @@ set -e
 eval "$(./conda/bin/conda shell.bash hook)"
 conda activate ./env
 
+yum makecache && yum install libglvnd-devel mesa-libGL mesa-libGL-devel mesa-libEGL mesa-libEGL-devel glfw mesa-libOSMesa-devel glew glew-devel egl-utils freeglut xorg-x11-server-Xvfb -y
+
 export PYTORCH_TEST_WITH_SLOW='1'
 python -m torch.utils.collect_env
 # Avoid error: "fatal: unsafe repository"

diff --git a/.circleci/unittest/linux_libs/scripts_habitat/install.sh b/.circleci/unittest/linux_libs/scripts_habitat/install.sh
@@ -41,7 +41,7 @@ fi
 pip install git+https://github.com/pytorch-labs/tensordict
 
 # smoke test
-python -c "import functorch"
+python -c "import functorch;import tensordict"
 
 printf "* Installing torchrl\n"
 pip3 install -e .

diff --git a/.circleci/unittest/linux_libs/scripts_jumanji/install.sh b/.circleci/unittest/linux_libs/scripts_jumanji/install.sh
@@ -35,8 +35,11 @@ else
     pip3 install --pre torch --extra-index-url https://download.pytorch.org/whl/nightly/cu116 --force-reinstall
 fi
 
+# install tensordict
+pip install git+https://github.com/pytorch-labs/tensordict
+
 # smoke test
-python -c "import functorch"
+python -c "import functorch;import tensordict"
 
 printf "* Installing torchrl\n"
 pip3 install -e .

diff --git a/.circleci/unittest/linux_olddeps/scripts_gym_0_13/install.sh b/.circleci/unittest/linux_olddeps/scripts_gym_0_13/install.sh
@@ -44,5 +44,9 @@ fi
 # install tensordict
 pip install git+https://github.com/pytorch-labs/tensordict
 
+# smoke test
+python -c "import tensordict"
+
 printf "* Installing torchrl\n"
 python setup.py develop
+python -c "import torchrl"