Skip to content

Commit

Permalink
Update on "[executorch][flat_tensor] DataMap implementation"
Browse files Browse the repository at this point in the history
DataMap implementation that
* Loads a flat_tensor file
* Populates a map with {fqn: tensor} and {fqn: TensorLayout}.
* Makes tensor information available via the named_data_map.h interface.

For now, DataMap doesn't store the DataLoader.
- If/when tensors are in their own segments, DataMap should also store a DataLoader.

Differential Revision: [D67064580](https://our.internmc.facebook.com/intern/diff/D67064580/)

[ghstack-poisoned]
  • Loading branch information
lucylq committed Feb 5, 2025
2 parents 9c37b57 + f0ac875 commit 7a8f1a6
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions extension/flat_tensor/targets.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,6 @@ def define_common_targets():
],
exported_headers = ["flat_tensor_data_map.h"],
deps = [
"//executorch/extension/flat_tensor/serialize:schema",
"//executorch/extension/flat_tensor/serialize:serialize",
"//executorch/extension/flat_tensor/serialize:generated_headers",
"//executorch/extension/flat_tensor/serialize:flat_tensor_header",
"//executorch/runtime/core:core",
Expand Down

0 comments on commit 7a8f1a6

Please sign in to comment.