Skip to content

Commit 002d68b

Browse files
committed
make handle droppable
1 parent be85a94 commit 002d68b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/forge/util/_shared_tensor.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,9 @@ def to_shared_tensor(self) -> SharedTensor:
3131
"""
3232
return SharedTensor(handle=self)
3333

34+
def drop(self) -> None:
35+
self.to_shared_tensor().drop()
36+
3437

3538
class SharedTensor:
3639
"""Wrapper class for tensors backed my shared memory"""

0 commit comments

Comments
 (0)