Skip to content

[core][gpu-objects] Support TensorDict #51550

@kevin85421

Description

@kevin85421

TensorDict uses a different serialization method from torch.Tensor. We should support extracting CUDA TensorDicts in the same way that we do for torch.Tensor.

Metadata

Metadata

Assignees

Labels

P1Issue that should be fixed within a few weekscommunity-backlogcoreIssues that should be addressed in Ray CoreenhancementRequest for new feature and/or capabilityrdtRay Direct Transportverl-poc

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions