| # RUN: env PYTHONPATH=%ir2vec_python_dir %python %s %S/../Inputs/input.ll %ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json | FileCheck %s | |
| import sys | |
| import ir2vec | |
| ll_file = sys.argv[1] | |
| vocab_path = sys.argv[2] | |
| vocab = ir2vec.loadVocab(vocab_path) | |
| tool = ir2vec.initEmbedding( | |
| filename=ll_file, mode=ir2vec.IR2VecKind.Symbolic, vocab=vocab | |
| ) | |
| # Success case | |
| emb = tool.getFuncEmb("add") | |
| print(f"SUCCESS: {emb.tolist()}") | |
| # CHECK: SUCCESS: [38.0, 40.0, 42.0] | |
| # Error: Function not found | |
| try: | |
| tool.getFuncEmb("nonexistent") | |
| except ValueError: | |
| print("ERROR: Function not found") | |
| # CHECK: ERROR: Function not found |