Skip to content

Commit 90ef7a9

Browse files
Revert "Supporting non-tensor-data write_size in planner write items. (pytorch#149434)"
This reverts commit 1442230. Reverted pytorch#149434 on behalf of https://github.com/izaitsevfb due to breaking docs build ([comment](pytorch#149434 (comment)))
1 parent 00333c4 commit 90ef7a9

File tree

1 file changed

+0
-9
lines changed

1 file changed

+0
-9
lines changed

torch/distributed/checkpoint/planner.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
__all__ = [
2121
"WriteItemType",
2222
"LoadItemType",
23-
"BytesIOWriteData",
2423
"TensorWriteData",
2524
"WriteItem",
2625
"ReadItem",
@@ -42,11 +41,6 @@ class LoadItemType(Enum):
4241
BYTE_IO = auto()
4342

4443

45-
@dataclass(frozen=True)
46-
class BytesIOWriteData:
47-
nbytes: int
48-
49-
5044
@dataclass(frozen=True)
5145
class TensorWriteData:
5246
chunk: ChunkStorageMetadata
@@ -61,9 +55,6 @@ class WriteItem:
6155
index: MetadataIndex
6256
type: WriteItemType
6357

64-
# Size of bytesIO data to be written.
65-
bytes_io_data: Optional[BytesIOWriteData] = None
66-
6758
# Value present if it's a tensor write
6859
tensor_data: Optional[TensorWriteData] = None
6960

0 commit comments

Comments
 (0)