[llvm] [mlgo-utils] Hoist entry script out to the correct directory (PR #146981)

via llvm-commits llvm-commits at lists.llvm.org
Thu Jul 3 18:29:04 PDT 2025


llvmbot wrote:


<!--LLVM PR SUMMARY COMMENT-->

@llvm/pr-subscribers-mlgo

Author: Vincent Lee (thevinster)

<details>
<summary>Changes</summary>

These scripts belong in the `mlgo-utils` directory when directly used with python3. Adjust the bazel paths to account for this as well. This follows the same structure as lit. 

```
$ python3 ./llvm/utils/mlgo-utils/extract_ir.py --help                                                                                                        1s
usage: extract_ir.py [-h] [--input INPUT] [--input_type [{json,params,directory,bazel_aquery}]] [--output_dir OUTPUT_DIR] [--num_workers [NUM_WORKERS]] [--llvm_objcopy_path [LLVM_OBJCOPY_PATH]]
                     [--obj_base_dir [OBJ_BASE_DIR]] [--cmd_filter [CMD_FILTER]] [--thinlto_build [{distributed,local}]] [--cmd_section_name [CMD_SECTION_NAME]]
                     [--bitcode_section_name [BITCODE_SECTION_NAME]] [--verbosity [{DEBUG,INFO,WARNING,ERROR}]]
...

$ python3 ./llvm/utils/mlgo-utils/combine_training_corpus.py --help                                                                                           0s
usage: combine_training_corpus.py [-h] [--root_dir ROOT_DIR] [--verbosity [{DEBUG,INFO,WARNING,ERROR}]]
...

$ python3 ./llvm/utils/mlgo-utils/make_corpus.py --help                                                                                                      0s
usage: make_corpus.py [-h] [--input_dir INPUT_DIR] [--output_dir OUTPUT_DIR] [--default_args [DEFAULT_ARGS]]
...
```

---
Full diff: https://github.com/llvm/llvm-project/pull/146981.diff


4 Files Affected:

- (renamed) llvm/utils/mlgo-utils/combine_training_corpus.py () 
- (renamed) llvm/utils/mlgo-utils/extract_ir.py () 
- (renamed) llvm/utils/mlgo-utils/make_corpus.py () 
- (modified) utils/bazel/llvm-project-overlay/llvm/BUILD.bazel (+3-3) 


``````````diff
diff --git a/llvm/utils/mlgo-utils/mlgo/corpus/combine_training_corpus.py b/llvm/utils/mlgo-utils/combine_training_corpus.py
similarity index 100%
rename from llvm/utils/mlgo-utils/mlgo/corpus/combine_training_corpus.py
rename to llvm/utils/mlgo-utils/combine_training_corpus.py
diff --git a/llvm/utils/mlgo-utils/mlgo/corpus/extract_ir.py b/llvm/utils/mlgo-utils/extract_ir.py
similarity index 100%
rename from llvm/utils/mlgo-utils/mlgo/corpus/extract_ir.py
rename to llvm/utils/mlgo-utils/extract_ir.py
diff --git a/llvm/utils/mlgo-utils/mlgo/corpus/make_corpus.py b/llvm/utils/mlgo-utils/make_corpus.py
similarity index 100%
rename from llvm/utils/mlgo-utils/mlgo/corpus/make_corpus.py
rename to llvm/utils/mlgo-utils/make_corpus.py
diff --git a/utils/bazel/llvm-project-overlay/llvm/BUILD.bazel b/utils/bazel/llvm-project-overlay/llvm/BUILD.bazel
index b618c74c19da1..db8a92fd25de6 100644
--- a/utils/bazel/llvm-project-overlay/llvm/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/llvm/BUILD.bazel
@@ -5210,8 +5210,8 @@ py_binary(
 py_binary(
     name = "extract_ir",
     srcs = [
+        "utils/mlgo-utils/extract_ir.py",
         "utils/mlgo-utils/mlgo/__init__.py",
-        "utils/mlgo-utils/mlgo/corpus/extract_ir.py",
         "utils/mlgo-utils/mlgo/corpus/extract_ir_lib.py",
         "utils/mlgo-utils/mlgo/corpus/flags.py",
     ],
@@ -5221,8 +5221,8 @@ py_binary(
 py_binary(
     name = "combine_training_corpus",
     srcs = [
+        "utils/mlgo-utils/combine_training_corpus.py",
         "utils/mlgo-utils/mlgo/__init__.py",
-        "utils/mlgo-utils/mlgo/corpus/combine_training_corpus.py",
         "utils/mlgo-utils/mlgo/corpus/combine_training_corpus_lib.py",
         "utils/mlgo-utils/mlgo/corpus/flags.py",
     ],
@@ -5232,8 +5232,8 @@ py_binary(
 py_binary(
     name = "make_corpus",
     srcs = [
+        "utils/mlgo-utils/make_corpus.py",
         "utils/mlgo-utils/mlgo/__init__.py",
-        "utils/mlgo-utils/mlgo/corpus/make_corpus.py",
         "utils/mlgo-utils/mlgo/corpus/make_corpus_lib.py",
     ],
     imports = ["utils/mlgo-utils"],

``````````

</details>


https://github.com/llvm/llvm-project/pull/146981


More information about the llvm-commits mailing list