Skip to content

Commit 8a0a35f

Browse files
Fix get_bbs docstring
The docstring only covered two of the arguments and a lot more have been added since the docstring was last updated. This patch updates the docstring to match the current argument set. Reviewers: orodley, virajbshah, ondrasej Reviewed By: ondrasej Pull Request: #338
1 parent d3790a8 commit 8a0a35f

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

gematria/datasets/pipelines/compile_modules_lib.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -250,6 +250,15 @@ def get_bbs(
250250
files to process.
251251
output_file: The output file pattern to use when writing the basic blocks
252252
to disk.
253+
remove_memory_accessing_instructions: Whether or not to remove memory
254+
accessing instructions when processing/filtering basic blocks.
255+
annotator_type: The annotator implementation to use to add memory
256+
annotations to the basic blocks.
257+
max_annotation_attempts: The maximum number of times the annotator should
258+
attempt to annotate the block before giving up.
259+
vocab_output_file: The output pattern for the vocabulary file.
260+
skip_no_loop_register: Whether or not to omit basic blocks for which a free
261+
register to use as a loop counter cannot be found.
253262
254263
Returns:
255264
A function that accepts a beam pipeline and adds on all the steps needed

0 commit comments

Comments
 (0)