Skip to content

v0.8.3: Better CudaGraphModule

Choose a tag to compare

@vmoens vmoens released this 16 May 15:24
· 123 commits to main since this release

This minor release provides some fixes to CudaGraphModule, allowing the module to run on different devices than the default.

It also adds __copy__ to the TensorDict ops, such that copy(td) triggers td.copy(), resulting in a copy of the TD stucture without new memory allocation.

Full Changelog: v0.8.2...v0.8.3