From e3c7f14c86e9f15eef3e02a6a0724b1dc4fd8d94 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Tue, 30 Jul 2024 21:03:56 -0400 Subject: [PATCH 1/3] init --- .ci/docker/requirements.txt | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt index 183867b2fcd..7cf3084e66f 100644 --- a/.ci/docker/requirements.txt +++ b/.ci/docker/requirements.txt @@ -28,9 +28,8 @@ tensorboard jinja2==3.1.3 pytorch-lightning torchx -# TODO: use stable 0.5 when released --e git+https://github.com/pytorch/rl.git#egg=torchrl --e git+https://github.com/pytorch/tensordict.git#egg=tensordict +torchrl==0.5.0 +tensordict==0.5.0 ax-platform nbformat>==5.9.2 datasets From 7137a78afbaad8c6961c4149b541d8c3a1222c90 Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Wed, 31 Jul 2024 15:04:34 -0700 Subject: [PATCH 2/3] Update .ci/docker/requirements.txt --- .ci/docker/requirements.txt | 3 --- 1 file changed, 3 deletions(-) diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt index 51659ebfca3..00cf2f21033 100644 --- a/.ci/docker/requirements.txt +++ b/.ci/docker/requirements.txt @@ -30,9 +30,6 @@ pytorch-lightning torchx torchrl==0.5.0 tensordict==0.5.0 -# TODO: use stable 0.5 when released --e git+https://github.com/pytorch/rl.git#egg=torchrl --e git+https://github.com/pytorch/tensordict.git#egg=tensordict ax-platform>==0.4.0 nbformat>==5.9.2 datasets From a04d09e8e6b3654e18427e27534da8cdd91db909 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Wed, 31 Jul 2024 18:45:59 -0400 Subject: [PATCH 3/3] fix after -> before --- intermediate_source/pinmem_nonblock.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/intermediate_source/pinmem_nonblock.py b/intermediate_source/pinmem_nonblock.py index 78e57b6cf1b..fa69507a0e7 100644 --- a/intermediate_source/pinmem_nonblock.py +++ b/intermediate_source/pinmem_nonblock.py @@ -556,7 +556,7 @@ def pin_copy_to_device_nonblocking(*tensors): # CUDA-enabled device. # # In summary, copying data from CPU to GPU is safe when using ``non_blocking=True``, but for any other direction, -# ``non_blocking=True`` can still be used but the user must make sure that a device synchronization is executed after +# ``non_blocking=True`` can still be used but the user must make sure that a device synchronization is executed before # the data is accessed. # # Practical recommendations