[MIR2Vec] Add embedder for machine instructions

[svkeerthy]

Implement MIR2Vec embedder for generating vector representations of Machine IR instructions, basic blocks, and functions. This patch introduces changes necessary to embed machine opcodes. Machine operands would be handled incrementally in the upcoming patches.

[svkeerthy]

• [MIR2Vec] Add MIR support to triplet generator script #164332
• [llvm-ir2vec][MIR2Vec] Supporting MIR mode in triplet and entity generation #164329
• [MLGO] Add MIR2Vec embedding documentation #164033
• [NFC][llvm-ir2vec] Standardize error message format using WithColor #164032
• [MIR2Vec][llvm-ir2vec] Add MIR2Vec support to llvm-ir2vec tool #164025
• [MIR2Vec] Handle Operands #163281
• [MIR2Vec] Add embedder for machine instructions #162161 👈 (View in Graphite)
• [MIR2Vec] Refactor MIR vocabulary to use opcode-based indexing #161713
• [IR2Vec] Initial infrastructure for MIR2Vec #161463
• [IR2Vec] Refactor vocabulary to use section-based storage #158376
• [IR2Vec] Add support for Cmp predicates in vocabulary and embeddings #156952
• [NFC][IR2Vec] Initialize Embedding vectors with zeros by default #155690
• [NFC][IR2Vec] Add usage documentation for obtaining embeddings #155516
• [IR2Vec] Refactor vocabulary to use canonical type IDs #155323 : 1 other dependent PR (#155700 )
• [NFC][IR2Vec] Add missed ptrtoaddr in vocab for tests #153094
• [IR2Vec] Make IR2VecCategory externally visible and reuse in llvm-ir2vec cl options #153089
• [IR2Vec][llvm-ir2vec] Supporting flow-aware embeddings #153087
• [IR2Vec] Add support for flow-aware embeddings #152613
• main

This stack of pull requests is managed by Graphite. Learn more about stacking.

[github-actions]

✅ With the latest revision this PR passed the C/C++ code formatter.

[svkeerthy]

Merge activity

• Oct 21, 5:12 PM UTC: A user started a stack merge that includes this pull request via Graphite.
• Oct 21, 5:14 PM UTC: @svkeerthy merged this pull request with Graphite.

[llvm-ci]

LLVM Buildbot has detected a new failure on builder llvm-clang-x86_64-win-fast running on as-builder-3 while building llvm at step 7 "test-build-unified-tree-check-llvm-unit".

Full details are available at: https://lab.llvm.org/buildbot/#/builders/2/builds/36602

Here is the relevant piece of the build log for the reference

Step 7 (test-build-unified-tree-check-llvm-unit) failure: test (failure)
******************** TEST 'LLVM-Unit :: CodeGen/./CodeGenTests.exe/73/74' FAILED ********************
Script(shard):
--
GTEST_OUTPUT=json:C:\buildbot\as-builder-3\llvm-clang-x86_64-win-fast\build\unittests\CodeGen\.\CodeGenTests.exe-LLVM-Unit-10904-73-74.json GTEST_SHUFFLE=0 GTEST_TOTAL_SHARDS=74 GTEST_SHARD_INDEX=73 C:\buildbot\as-builder-3\llvm-clang-x86_64-win-fast\build\unittests\CodeGen\.\CodeGenTests.exe
--

Script:
--
C:\buildbot\as-builder-3\llvm-clang-x86_64-win-fast\build\unittests\CodeGen\.\CodeGenTests.exe --gtest_filter=MIR2VecEmbeddingTestFixture.TestSymbolicEmbedder
--
C:\buildbot\as-builder-3\llvm-clang-x86_64-win-fast\llvm-project\llvm\unittests\CodeGen\MIR2VecTest.cpp(68): Skipped
x86_64-unknown-linux-gnu target triple not available; Skipping test

unknown file: error: SEH exception with code 0x3221225477 thrown in SetUp().
Stack trace:



unknown file
SEH exception with code 0x3221225477 thrown in SetUp().
Stack trace:




********************

[llvm-ci]

LLVM Buildbot has detected a new failure on builder llvm-clang-aarch64-darwin running on doug-worker-4 while building llvm at step 6 "test-build-unified-tree-check-all".

Full details are available at: https://lab.llvm.org/buildbot/#/builders/190/builds/29470

Here is the relevant piece of the build log for the reference

Step 6 (test-build-unified-tree-check-all) failure: test (failure)
******************** TEST 'LLVM-Unit :: CodeGen/./CodeGenTests/0/19' FAILED ********************
Script(shard):
--
GTEST_OUTPUT=json:/Volumes/RAMDisk/buildbot-root/aarch64-darwin/build/unittests/CodeGen/./CodeGenTests-LLVM-Unit-69057-0-19.json GTEST_SHUFFLE=0 GTEST_TOTAL_SHARDS=19 GTEST_SHARD_INDEX=0 /Volumes/RAMDisk/buildbot-root/aarch64-darwin/build/unittests/CodeGen/./CodeGenTests
--

Note: This is test shard 1 of 19.
[==========] Running 16 tests from 15 test suites.
[----------] Global test environment set-up.
[----------] 1 test from AllocationOrderTest
[ RUN      ] AllocationOrderTest.Basic
[       OK ] AllocationOrderTest.Basic (0 ms)
[----------] 1 test from AllocationOrderTest (0 ms total)

[----------] 1 test from AsmPrinterEmitDwarfStringOffsetTest
[ RUN      ] AsmPrinterEmitDwarfStringOffsetTest.DWARF64
/Users/buildbot/buildbot-root/aarch64-darwin/llvm-project/llvm/unittests/CodeGen/AsmPrinterDwarfTest.cpp:180: Skipped


[  SKIPPED ] AsmPrinterEmitDwarfStringOffsetTest.DWARF64 (0 ms)
[----------] 1 test from AsmPrinterEmitDwarfStringOffsetTest (0 ms total)

[----------] 1 test from DIEHashTest
[ RUN      ] DIEHashTest.ReusedType
[       OK ] DIEHashTest.ReusedType (0 ms)
[----------] 1 test from DIEHashTest (0 ms total)

[----------] 1 test from DroppedVariableStatsMIR
[ RUN      ] DroppedVariableStatsMIR.InlinedAtShared
[       OK ] DroppedVariableStatsMIR.InlinedAtShared (6 ms)
[----------] 1 test from DroppedVariableStatsMIR (6 ms total)

[----------] 1 test from InstrRefLDVTest
[ RUN      ] InstrRefLDVTest.pickVPHILocBadlyNestedLoops
/Users/buildbot/buildbot-root/aarch64-darwin/llvm-project/llvm/unittests/CodeGen/InstrRefLDVTest.cpp:81: Skipped


[  SKIPPED ] InstrRefLDVTest.pickVPHILocBadlyNestedLoops (0 ms)
[----------] 1 test from InstrRefLDVTest (0 ms total)

[----------] 1 test from LowLevelTypeTest
[ RUN      ] LowLevelTypeTest.ConstExpr
[       OK ] LowLevelTypeTest.ConstExpr (0 ms)
[----------] 1 test from LowLevelTypeTest (0 ms total)

[----------] 1 test from IsIdenticalToTest
[ RUN      ] IsIdenticalToTest.DifferentDefs
[       OK ] IsIdenticalToTest.DifferentDefs (0 ms)
[----------] 1 test from IsIdenticalToTest (0 ms total)

...