Skip to content

Commit b61a717

Browse files
authored
[1/N][TRTLLM-5195][feat] Share PyTorch tensor between processes (NVIDIA#5396)
1 parent 3f7cede commit b61a717

File tree

3 files changed

+687
-0
lines changed

3 files changed

+687
-0
lines changed
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
from .shared_tensor import (SharedTensorContainer,
2+
_SharedTensorRebuildMethodRegistry)
3+
4+
# Initialize the registry when the package is imported
5+
_SharedTensorRebuildMethodRegistry.initialize()
6+
7+
__all__ = [
8+
'SharedTensorContainer',
9+
]

0 commit comments

Comments
 (0)