-
Notifications
You must be signed in to change notification settings - Fork 11
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add support for consuming instruction annotations to TokenGraphBuilderModel
s.
#94
Add support for consuming instruction annotations to TokenGraphBuilderModel
s.
#94
Commits on Mar 26, 2024
-
Add instruction annotation message to proto specification.
* Add new message type `AnnotationProto` (`gematria/proto/annotation.proto`). * Include it `CanonicalizedInstructionProto`.
Configuration menu - View commit details
-
Copy full SHA for fe0b7f3 - Browse repository at this point
Copy the full SHA fe0b7f3View commit details -
Add instruction annotation support to basic block representation.
* Add new `Annotation` class. * Add `instruction_annotations` vector to `Instruction`. * Add Python bindings for these. * Add tests for all above changes.
Configuration menu - View commit details
-
Copy full SHA for 63a914f - Browse repository at this point
Copy the full SHA 63a914fView commit details -
Add conversions between
Annotation
andAnnotationProto
and back.* Add `Annotation` conversion and update `Instruction` conversion. * Refactor `ToVector` and `ToRepeatedPtrField` to be generic. * Add Python bindings for conversions. * Add tests for all changes.
Configuration menu - View commit details
-
Copy full SHA for aafb571 - Browse repository at this point
Copy the full SHA aafb571View commit details -
Store instruction annotations while building graph.
* Make instruction annotations available to models via the graph builder, where they are stored as a vector-of-vector matrix. * Add Python bindings for the same. * Add tests for the C++ graph builder implementation, and update the tests for the Python bindings to be compatible with the changes.
Configuration menu - View commit details
-
Copy full SHA for 160e854 - Browse repository at this point
Copy the full SHA 160e854View commit details -
Add test for graph builder instruction annotation Python bindings.
* Add a test adding basic blocks with annotations to the graph builder. * Add `annotated_basic_blocks_with_throughput.pbtxt` as a source for annotated basic blocks for tests. * Add utilities to get the annotated blocks to `gematria/testing`.
Configuration menu - View commit details
-
Copy full SHA for 1c9e599 - Browse repository at this point
Copy the full SHA 1c9e599View commit details -
Incorporate instruction annotations in node embedding computation.
* The ends of node embedding vectors used by `TokenGraphBuilderModel`s for instructions are overwritten by the corresponding instruction annotations. * A test for `TokenGraphBuilderModel`s consuming instruction annotations has been added as well.
Configuration menu - View commit details
-
Copy full SHA for 8527839 - Browse repository at this point
Copy the full SHA 8527839View commit details -
Add command-line support for running models consuming annotations.
* Add a `--gematria_annotation_names_file` flag to `run_granite_model` to pass in a list of instruction annotation names consumed by the model.
Configuration menu - View commit details
-
Copy full SHA for a7c333e - Browse repository at this point
Copy the full SHA a7c333eView commit details
Commits on Jun 17, 2024
-
Configuration menu - View commit details
-
Copy full SHA for c3c95aa - Browse repository at this point
Copy the full SHA c3c95aaView commit details
Commits on Jun 19, 2024
-
Configuration menu - View commit details
-
Copy full SHA for 13f4db9 - Browse repository at this point
Copy the full SHA 13f4db9View commit details
Commits on Jun 21, 2024
-
* Move annotation-related tensors from `TokenGraphBuilderModel` to `GraphBuilderModelBase`. * Adjust the order in which annotation-related tensors are created relative to other parts of the graph. * Refactor `TokenGraphBuilderModelNodeEmbed`.
Configuration menu - View commit details
-
Copy full SHA for 29a5c11 - Browse repository at this point
Copy the full SHA 29a5c11View commit details
Commits on Jul 18, 2024
-
Configuration menu - View commit details
-
Copy full SHA for ff2fcf5 - Browse repository at this point
Copy the full SHA ff2fcf5View commit details
Commits on Jul 22, 2024
-
Configuration menu - View commit details
-
Copy full SHA for 58eab6f - Browse repository at this point
Copy the full SHA 58eab6fView commit details -
Configuration menu - View commit details
-
Copy full SHA for 0fe7a7f - Browse repository at this point
Copy the full SHA 0fe7a7fView commit details -
Temporarily pin TensorFlow version to 2.15
* TF 2.15 is the last version to support Keras 2 without any fixes. * Until google#132 lands with a fix that lets us keep using Keras 2, this pins the TF version to 2.15 for CI. * After this lands, google#132 can be updated to also patch the Keras 2 usage in this PR and then we can bump the TF version up.
Configuration menu - View commit details
-
Copy full SHA for f4b0da1 - Browse repository at this point
Copy the full SHA f4b0da1View commit details