Skip to content

Commit 7436967

Browse files
committed
Update on "[executorch][flat_tensor] Serialize flat tensor"
Serialize a flat tensor file. The resulting file looks like: Header with - flatbuffer offset and size - segment data offset and size Flatbuffer Tensor data (in segment) Differential Revision: [D66374253](https://our.internmc.facebook.com/intern/diff/D66374253/) [ghstack-poisoned]
2 parents a337c10 + 462ac05 commit 7436967

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

extension/flat_tensor/test/test_serialize.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -64,16 +64,16 @@ def test_serialize(self) -> None:
6464
header.flatbuffer_offset,
6565
aligned_size(FlatTensorHeader.EXPECTED_LENGTH, config.segment_alignment),
6666
)
67-
67+
6868
# Flatbuffer is non-empty.
6969
self.assertTrue(header.flatbuffer_size > 0)
70-
70+
7171
# Segment base offset is aligned to config.segment_alignment.
7272
expected_segment_base_offset = aligned_size(
7373
header.flatbuffer_offset + header.flatbuffer_size, config.segment_alignment
7474
)
7575
self.assertTrue(header.segment_base_offset, expected_segment_base_offset)
76-
76+
7777
# TEST_TENSOR_BUFFER is aligned to config.segment_alignment.
7878
self.assertEqual(header.segment_data_size, config.segment_alignment)
7979

0 commit comments

Comments
 (0)