diff --git a/llvm/test/tools/llvm-ir2vec/embeddings-flowaware.ll b/llvm/test/tools/llvm-ir2vec/embeddings-flowaware.ll index ade228d7a3665..310f79f3d4303 100644 --- a/llvm/test/tools/llvm-ir2vec/embeddings-flowaware.ll +++ b/llvm/test/tools/llvm-ir2vec/embeddings-flowaware.ll @@ -1,73 +1,151 @@ ; RUN: llvm-ir2vec embeddings --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-DEFAULT -; RUN: llvm-ir2vec embeddings --level=func --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-FUNC-LEVEL -; RUN: llvm-ir2vec embeddings --level=func --function=abc --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-FUNC-LEVEL-ABC -; RUN: not llvm-ir2vec embeddings --level=func --function=def --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s 2>&1 | FileCheck %s -check-prefix=CHECK-FUNC-DEF -; RUN: llvm-ir2vec embeddings --level=bb --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-BB-LEVEL -; RUN: llvm-ir2vec embeddings --level=bb --function=abc_repeat --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-BB-LEVEL-ABC-REPEAT -; RUN: llvm-ir2vec embeddings --level=inst --function=abc_repeat --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-INST-LEVEL-ABC-REPEAT - -define dso_local noundef float @abc(i32 noundef %a, float noundef %b) #0 { -entry: - %a.addr = alloca i32, align 4 - %b.addr = alloca float, align 4 - store i32 %a, ptr %a.addr, align 4 - store float %b, ptr %b.addr, align 4 - %0 = load i32, ptr %a.addr, align 4 - %1 = load i32, ptr %a.addr, align 4 - %mul = mul nsw i32 %0, %1 - %conv = sitofp i32 %mul to float - %2 = load float, ptr %b.addr, align 4 - %add = fadd float %conv, %2 - ret float %add +; RUN: llvm-ir2vec embeddings --level=func --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-FUNC-LEVEL + +; Test with mangled names +; RUN: llvm-ir2vec embeddings --level=func --function=_Z3addii --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-ADD-INT-INT +; RUN: llvm-ir2vec embeddings --level=func --function=_Z3addiii --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-ADD-INT-INT-INT +; RUN: llvm-ir2vec embeddings --level=func --function=_Z3adddd --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-ADD-DOUBLE-DOUBLE +; RUN: llvm-ir2vec embeddings --level=func --function=main --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-MAIN + +; Test with demangled names +; RUN: llvm-ir2vec embeddings --level=func --function="add(int, int)" --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-DEMANGLED-INT-INT +; RUN: llvm-ir2vec embeddings --level=func --function="add(int, int, int)" --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-DEMANGLED-INT-INT-INT +; RUN: llvm-ir2vec embeddings --level=func --function="add(double, double)" --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-DEMANGLED-DOUBLE-DOUBLE + +; Test basic block level for all functions +; RUN: llvm-ir2vec embeddings --level=bb --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-BB-LEVEL-ALL + +; Test basic block level for one function +; RUN: llvm-ir2vec embeddings --level=bb --function=_Z3addii --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-BB-ADD-INT-INT + +; Test instruction level for one function +; RUN: llvm-ir2vec embeddings --level=inst --function=_Z3addiii --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-INST-ADD-INT-INT-INT + +; Test error case - non-existent function +; RUN: not llvm-ir2vec embeddings --level=func --function=_Z3subii --ir2vec-kind=flow-aware --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s 2>&1 | FileCheck %s -check-prefix=CHECK-NONEXISTENT + +; Function Attrs: mustprogress noinline nounwind optnone uwtable +; add(int, int) +define dso_local noundef i32 @_Z3addii(i32 noundef %0, i32 noundef %1) #0 { + %3 = alloca i32, align 4 + %4 = alloca i32, align 4 + store i32 %0, ptr %3, align 4 + store i32 %1, ptr %4, align 4 + %5 = load i32, ptr %3, align 4 + %6 = load i32, ptr %4, align 4 + %7 = add nsw i32 %5, %6 + ret i32 %7 +} + +; Function Attrs: mustprogress noinline nounwind optnone uwtable +; add(int, int, int) +define dso_local noundef i32 @_Z3addiii(i32 noundef %0, i32 noundef %1, i32 noundef %2) #0 { + %4 = alloca i32, align 4 + %5 = alloca i32, align 4 + %6 = alloca i32, align 4 + store i32 %0, ptr %4, align 4 + store i32 %1, ptr %5, align 4 + store i32 %2, ptr %6, align 4 + %7 = load i32, ptr %4, align 4 + %8 = load i32, ptr %5, align 4 + %9 = add nsw i32 %7, %8 + %10 = load i32, ptr %6, align 4 + %11 = add nsw i32 %9, %10 + ret i32 %11 } -define dso_local noundef float @abc_repeat(i32 noundef %a, float noundef %b) #0 { -entry: - %a.addr = alloca i32, align 4 - %b.addr = alloca float, align 4 - store i32 %a, ptr %a.addr, align 4 - store float %b, ptr %b.addr, align 4 - %0 = load i32, ptr %a.addr, align 4 - %1 = load i32, ptr %a.addr, align 4 - %mul = mul nsw i32 %0, %1 - %conv = sitofp i32 %mul to float - %2 = load float, ptr %b.addr, align 4 - %add = fadd float %conv, %2 - ret float %add +; Function Attrs: mustprogress noinline nounwind optnone uwtable +; add(double, double) +define dso_local noundef double @_Z3adddd(double noundef %0, double noundef %1) #0 { + %3 = alloca double, align 8 + %4 = alloca double, align 8 + store double %0, ptr %3, align 8 + store double %1, ptr %4, align 8 + %5 = load double, ptr %3, align 8 + %6 = load double, ptr %4, align 8 + %7 = fadd double %5, %6 + ret double %7 } -; CHECK-DEFAULT: Function: abc -; CHECK-DEFAULT-NEXT: [ 3630.00 3672.00 3714.00 ] -; CHECK-DEFAULT-NEXT: Function: abc_repeat -; CHECK-DEFAULT-NEXT: [ 3630.00 3672.00 3714.00 ] - -; CHECK-FUNC-LEVEL: Function: abc -; CHECK-FUNC-LEVEL-NEXT: [ 3630.00 3672.00 3714.00 ] -; CHECK-FUNC-LEVEL-NEXT: Function: abc_repeat -; CHECK-FUNC-LEVEL-NEXT: [ 3630.00 3672.00 3714.00 ] - -; CHECK-FUNC-LEVEL-ABC: Function: abc -; CHECK-FUNC-LEVEL-NEXT-ABC: [ 3630.00 3672.00 3714.00 ] - -; CHECK-FUNC-DEF: error: Function 'def' not found - -; CHECK-BB-LEVEL: Function: abc -; CHECK-BB-LEVEL-NEXT: entry: [ 3630.00 3672.00 3714.00 ] -; CHECK-BB-LEVEL-NEXT: Function: abc_repeat -; CHECK-BB-LEVEL-NEXT: entry: [ 3630.00 3672.00 3714.00 ] - -; CHECK-BB-LEVEL-ABC-REPEAT: Function: abc_repeat -; CHECK-BB-LEVEL-ABC-REPEAT-NEXT: entry: [ 3630.00 3672.00 3714.00 ] - -; CHECK-INST-LEVEL-ABC-REPEAT: Function: abc_repeat -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %a.addr = alloca i32, align 4 [ 91.00 92.00 93.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %b.addr = alloca float, align 4 [ 91.00 92.00 93.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: store i32 %a, ptr %a.addr, align 4 [ 188.00 190.00 192.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: store float %b, ptr %b.addr, align 4 [ 188.00 190.00 192.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %0 = load i32, ptr %a.addr, align 4 [ 185.00 187.00 189.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %1 = load i32, ptr %a.addr, align 4 [ 185.00 187.00 189.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %mul = mul nsw i32 %0, %1 [ 419.00 424.00 429.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %conv = sitofp i32 %mul to float [ 549.00 555.00 561.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %2 = load float, ptr %b.addr, align 4 [ 185.00 187.00 189.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %add = fadd float %conv, %2 [ 774.00 783.00 792.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: ret float %add [ 775.00 785.00 795.00 ] +; Function Attrs: mustprogress noinline norecurse nounwind optnone uwtable +define dso_local noundef i32 @main() #1 { + %1 = alloca i32, align 4 + %2 = alloca i32, align 4 + %3 = alloca i32, align 4 + %4 = alloca double, align 8 + store i32 0, ptr %1, align 4 + %5 = call noundef i32 @_Z3addii(i32 noundef 5, i32 noundef 3) + store i32 %5, ptr %2, align 4 + %6 = call noundef i32 @_Z3addiii(i32 noundef 5, i32 noundef 3, i32 noundef 2) + store i32 %6, ptr %3, align 4 + %7 = call noundef double @_Z3adddd(double noundef 5.500000e+00, double noundef 3.200000e+00) + store double %7, ptr %4, align 8 + ret i32 0 +} + +; CHECK-DEFAULT: Function: _Z3addii +; CHECK-DEFAULT-NEXT: [ 1743.00 1764.00 1785.00 ] +; CHECK-DEFAULT-NEXT: Function: _Z3addiii +; CHECK-DEFAULT-NEXT: [ 3058.00 3095.00 3132.00 ] +; CHECK-DEFAULT-NEXT: Function: _Z3adddd +; CHECK-DEFAULT-NEXT: [ 1749.00 1770.00 1791.00 ] +; CHECK-DEFAULT-NEXT: Function: main +; CHECK-DEFAULT-NEXT: [ 2113.00 2132.00 2151.00 ] + +; CHECK-FUNC-LEVEL: Function: _Z3addii +; CHECK-FUNC-LEVEL-NEXT: [ 1743.00 1764.00 1785.00 ] +; CHECK-FUNC-LEVEL-NEXT: Function: _Z3addiii +; CHECK-FUNC-LEVEL-NEXT: [ 3058.00 3095.00 3132.00 ] +; CHECK-FUNC-LEVEL-NEXT: Function: _Z3adddd +; CHECK-FUNC-LEVEL-NEXT: [ 1749.00 1770.00 1791.00 ] +; CHECK-FUNC-LEVEL-NEXT: Function: main +; CHECK-FUNC-LEVEL-NEXT: [ 2113.00 2132.00 2151.00 ] + +; CHECK-ADD-INT-INT: Function: _Z3addii +; CHECK-ADD-INT-INT-NEXT: [ 1743.00 1764.00 1785.00 ] + +; CHECK-ADD-INT-INT-INT: Function: _Z3addiii +; CHECK-ADD-INT-INT-INT-NEXT: [ 3058.00 3095.00 3132.00 ] + +; CHECK-ADD-DOUBLE-DOUBLE: Function: _Z3adddd +; CHECK-ADD-DOUBLE-DOUBLE-NEXT: [ 1749.00 1770.00 1791.00 ] + +; CHECK-MAIN: Function: main +; CHECK-MAIN-NEXT: [ 2113.00 2132.00 2151.00 ] + +; CHECK-DEMANGLED-INT-INT: Function: _Z3addii +; CHECK-DEMANGLED-INT-INT-NEXT: [ 1743.00 1764.00 1785.00 ] + +; CHECK-DEMANGLED-INT-INT-INT: Function: _Z3addiii +; CHECK-DEMANGLED-INT-INT-INT-NEXT: [ 3058.00 3095.00 3132.00 ] + +; CHECK-DEMANGLED-DOUBLE-DOUBLE: Function: _Z3adddd +; CHECK-DEMANGLED-DOUBLE-DOUBLE-NEXT: [ 1749.00 1770.00 1791.00 ] + +; CHECK-BB-LEVEL-ALL: Function: _Z3addii +; CHECK-BB-LEVEL-ALL-NEXT: [ 1743.00 1764.00 1785.00 ] +; CHECK-BB-LEVEL-ALL-NEXT: Function: _Z3addiii +; CHECK-BB-LEVEL-ALL-NEXT: [ 3058.00 3095.00 3132.00 ] +; CHECK-BB-LEVEL-ALL-NEXT: Function: _Z3adddd +; CHECK-BB-LEVEL-ALL-NEXT: [ 1749.00 1770.00 1791.00 ] +; CHECK-BB-LEVEL-ALL-NEXT: Function: main +; CHECK-BB-LEVEL-ALL-NEXT: [ 2113.00 2132.00 2151.00 ] + +; CHECK-BB-ADD-INT-INT: Function: _Z3addii +; CHECK-BB-ADD-INT-INT-NEXT: [ 1743.00 1764.00 1785.00 ] + +; CHECK-INST-ADD-INT-INT-INT: Function: _Z3addiii +; CHECK-INST-ADD-INT-INT-INT-NEXT: %4 = alloca i32, align 4 [ 91.00 92.00 93.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %5 = alloca i32, align 4 [ 91.00 92.00 93.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %6 = alloca i32, align 4 [ 91.00 92.00 93.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: store i32 %0, ptr %4, align 4 [ 188.00 190.00 192.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: store i32 %1, ptr %5, align 4 [ 188.00 190.00 192.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: store i32 %2, ptr %6, align 4 [ 188.00 190.00 192.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %7 = load i32, ptr %4, align 4 [ 185.00 187.00 189.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %8 = load i32, ptr %5, align 4 [ 185.00 187.00 189.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %9 = add nsw i32 %7, %8 [ 407.00 412.00 417.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %10 = load i32, ptr %6, align 4 [ 185.00 187.00 189.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %11 = add nsw i32 %9, %10 [ 629.00 637.00 645.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: ret i32 %11 [ 630.00 639.00 648.00 ] + +; CHECK-NONEXISTENT: error: Function '_Z3subii' not found \ No newline at end of file diff --git a/llvm/test/tools/llvm-ir2vec/embeddings-symbolic.ll b/llvm/test/tools/llvm-ir2vec/embeddings-symbolic.ll index 9d60e12e8ab90..4eac0c3f9e46d 100644 --- a/llvm/test/tools/llvm-ir2vec/embeddings-symbolic.ll +++ b/llvm/test/tools/llvm-ir2vec/embeddings-symbolic.ll @@ -1,73 +1,151 @@ ; RUN: llvm-ir2vec embeddings --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-DEFAULT ; RUN: llvm-ir2vec embeddings --level=func --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-FUNC-LEVEL -; RUN: llvm-ir2vec embeddings --level=func --function=abc --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-FUNC-LEVEL-ABC -; RUN: not llvm-ir2vec embeddings --level=func --function=def --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s 2>&1 | FileCheck %s -check-prefix=CHECK-FUNC-DEF -; RUN: llvm-ir2vec embeddings --level=bb --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-BB-LEVEL -; RUN: llvm-ir2vec embeddings --level=bb --function=abc_repeat --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-BB-LEVEL-ABC-REPEAT -; RUN: llvm-ir2vec embeddings --level=inst --function=abc_repeat --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-INST-LEVEL-ABC-REPEAT - -define dso_local noundef float @abc(i32 noundef %a, float noundef %b) #0 { -entry: - %a.addr = alloca i32, align 4 - %b.addr = alloca float, align 4 - store i32 %a, ptr %a.addr, align 4 - store float %b, ptr %b.addr, align 4 - %0 = load i32, ptr %a.addr, align 4 - %1 = load i32, ptr %a.addr, align 4 - %mul = mul nsw i32 %0, %1 - %conv = sitofp i32 %mul to float - %2 = load float, ptr %b.addr, align 4 - %add = fadd float %conv, %2 - ret float %add + +; Test with mangled names +; RUN: llvm-ir2vec embeddings --level=func --function=_Z3addii --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-ADD-INT-INT +; RUN: llvm-ir2vec embeddings --level=func --function=_Z3addiii --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-ADD-INT-INT-INT +; RUN: llvm-ir2vec embeddings --level=func --function=_Z3adddd --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-ADD-DOUBLE-DOUBLE +; RUN: llvm-ir2vec embeddings --level=func --function=main --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-MAIN + +; Test with demangled names +; RUN: llvm-ir2vec embeddings --level=func --function="add(int, int)" --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-DEMANGLED-INT-INT +; RUN: llvm-ir2vec embeddings --level=func --function="add(int, int, int)" --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-DEMANGLED-INT-INT-INT +; RUN: llvm-ir2vec embeddings --level=func --function="add(double, double)" --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-DEMANGLED-DOUBLE-DOUBLE + +; Test basic block level for all functions +; RUN: llvm-ir2vec embeddings --level=bb --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-BB-LEVEL-ALL + +; Test basic block level for one function +; RUN: llvm-ir2vec embeddings --level=bb --function=_Z3addii --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-BB-ADD-INT-INT + +; Test instruction level for one function +; RUN: llvm-ir2vec embeddings --level=inst --function=_Z3addiii --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-INST-ADD-INT-INT-INT + +; Test error case - non-existent function +; RUN: not llvm-ir2vec embeddings --level=func --function=_Z3subii --ir2vec-vocab-path=%ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json %s 2>&1 | FileCheck %s -check-prefix=CHECK-NONEXISTENT + +; Function Attrs: mustprogress noinline nounwind optnone uwtable +; add(int, int) +define dso_local noundef i32 @_Z3addii(i32 noundef %0, i32 noundef %1) #0 { + %3 = alloca i32, align 4 + %4 = alloca i32, align 4 + store i32 %0, ptr %3, align 4 + store i32 %1, ptr %4, align 4 + %5 = load i32, ptr %3, align 4 + %6 = load i32, ptr %4, align 4 + %7 = add nsw i32 %5, %6 + ret i32 %7 +} + +; Function Attrs: mustprogress noinline nounwind optnone uwtable +; add(int, int, int) +define dso_local noundef i32 @_Z3addiii(i32 noundef %0, i32 noundef %1, i32 noundef %2) #0 { + %4 = alloca i32, align 4 + %5 = alloca i32, align 4 + %6 = alloca i32, align 4 + store i32 %0, ptr %4, align 4 + store i32 %1, ptr %5, align 4 + store i32 %2, ptr %6, align 4 + %7 = load i32, ptr %4, align 4 + %8 = load i32, ptr %5, align 4 + %9 = add nsw i32 %7, %8 + %10 = load i32, ptr %6, align 4 + %11 = add nsw i32 %9, %10 + ret i32 %11 } -define dso_local noundef float @abc_repeat(i32 noundef %a, float noundef %b) #0 { -entry: - %a.addr = alloca i32, align 4 - %b.addr = alloca float, align 4 - store i32 %a, ptr %a.addr, align 4 - store float %b, ptr %b.addr, align 4 - %0 = load i32, ptr %a.addr, align 4 - %1 = load i32, ptr %a.addr, align 4 - %mul = mul nsw i32 %0, %1 - %conv = sitofp i32 %mul to float - %2 = load float, ptr %b.addr, align 4 - %add = fadd float %conv, %2 - ret float %add +; Function Attrs: mustprogress noinline nounwind optnone uwtable +; add(double, double) +define dso_local noundef double @_Z3adddd(double noundef %0, double noundef %1) #0 { + %3 = alloca double, align 8 + %4 = alloca double, align 8 + store double %0, ptr %3, align 8 + store double %1, ptr %4, align 8 + %5 = load double, ptr %3, align 8 + %6 = load double, ptr %4, align 8 + %7 = fadd double %5, %6 + ret double %7 } -; CHECK-DEFAULT: Function: abc -; CHECK-DEFAULT-NEXT: [ 878.00 889.00 900.00 ] -; CHECK-DEFAULT-NEXT: Function: abc_repeat -; CHECK-DEFAULT-NEXT: [ 878.00 889.00 900.00 ] - -; CHECK-FUNC-LEVEL: Function: abc -; CHECK-FUNC-LEVEL-NEXT: [ 878.00 889.00 900.00 ] -; CHECK-FUNC-LEVEL-NEXT: Function: abc_repeat -; CHECK-FUNC-LEVEL-NEXT: [ 878.00 889.00 900.00 ] - -; CHECK-FUNC-LEVEL-ABC: Function: abc -; CHECK-FUNC-LEVEL-NEXT-ABC: [ 878.00 889.00 900.00 ] - -; CHECK-FUNC-DEF: error: Function 'def' not found - -; CHECK-BB-LEVEL: Function: abc -; CHECK-BB-LEVEL-NEXT: entry: [ 878.00 889.00 900.00 ] -; CHECK-BB-LEVEL-NEXT: Function: abc_repeat -; CHECK-BB-LEVEL-NEXT: entry: [ 878.00 889.00 900.00 ] - -; CHECK-BB-LEVEL-ABC-REPEAT: Function: abc_repeat -; CHECK-BB-LEVEL-ABC-REPEAT-NEXT: entry: [ 878.00 889.00 900.00 ] - -; CHECK-INST-LEVEL-ABC-REPEAT: Function: abc_repeat -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %a.addr = alloca i32, align 4 [ 91.00 92.00 93.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %b.addr = alloca float, align 4 [ 91.00 92.00 93.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: store i32 %a, ptr %a.addr, align 4 [ 97.00 98.00 99.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: store float %b, ptr %b.addr, align 4 [ 97.00 98.00 99.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %0 = load i32, ptr %a.addr, align 4 [ 94.00 95.00 96.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %1 = load i32, ptr %a.addr, align 4 [ 94.00 95.00 96.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %mul = mul nsw i32 %0, %1 [ 49.00 50.00 51.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %conv = sitofp i32 %mul to float [ 130.00 131.00 132.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %2 = load float, ptr %b.addr, align 4 [ 94.00 95.00 96.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: %add = fadd float %conv, %2 [ 40.00 41.00 42.00 ] -; CHECK-INST-LEVEL-ABC-REPEAT-NEXT: ret float %add [ 1.00 2.00 3.00 ] +; Function Attrs: mustprogress noinline norecurse nounwind optnone uwtable +define dso_local noundef i32 @main() #1 { + %1 = alloca i32, align 4 + %2 = alloca i32, align 4 + %3 = alloca i32, align 4 + %4 = alloca double, align 8 + store i32 0, ptr %1, align 4 + %5 = call noundef i32 @_Z3addii(i32 noundef 5, i32 noundef 3) + store i32 %5, ptr %2, align 4 + %6 = call noundef i32 @_Z3addiii(i32 noundef 5, i32 noundef 3, i32 noundef 2) + store i32 %6, ptr %3, align 4 + %7 = call noundef double @_Z3adddd(double noundef 5.500000e+00, double noundef 3.200000e+00) + store double %7, ptr %4, align 8 + ret i32 0 +} + +; CHECK-DEFAULT: Function: _Z3addii +; CHECK-DEFAULT-NEXT: [ 602.00 610.00 618.00 ] +; CHECK-DEFAULT-NEXT: Function: _Z3addiii +; CHECK-DEFAULT-NEXT: [ 921.00 933.00 945.00 ] +; CHECK-DEFAULT-NEXT: Function: _Z3adddd +; CHECK-DEFAULT-NEXT: [ 605.00 613.00 621.00 ] +; CHECK-DEFAULT-NEXT: Function: main +; CHECK-DEFAULT-NEXT: [ 1251.00 1263.00 1275.00 ] + +; CHECK-FUNC-LEVEL: Function: _Z3addii +; CHECK-FUNC-LEVEL-NEXT: [ 602.00 610.00 618.00 ] +; CHECK-FUNC-LEVEL-NEXT: Function: _Z3addiii +; CHECK-FUNC-LEVEL-NEXT: [ 921.00 933.00 945.00 ] +; CHECK-FUNC-LEVEL-NEXT: Function: _Z3adddd +; CHECK-FUNC-LEVEL-NEXT: [ 605.00 613.00 621.00 ] +; CHECK-FUNC-LEVEL-NEXT: Function: main +; CHECK-FUNC-LEVEL-NEXT: [ 1251.00 1263.00 1275.00 ] + +; CHECK-ADD-INT-INT: Function: _Z3addii +; CHECK-ADD-INT-INT-NEXT: [ 602.00 610.00 618.00 ] + +; CHECK-ADD-INT-INT-INT: Function: _Z3addiii +; CHECK-ADD-INT-INT-INT-NEXT: [ 921.00 933.00 945.00 ] + +; CHECK-ADD-DOUBLE-DOUBLE: Function: _Z3adddd +; CHECK-ADD-DOUBLE-DOUBLE-NEXT: [ 605.00 613.00 621.00 ] + +; CHECK-MAIN: Function: main +; CHECK-MAIN-NEXT: [ 1251.00 1263.00 1275.00 ] + +; CHECK-DEMANGLED-INT-INT: Function: _Z3addii +; CHECK-DEMANGLED-INT-INT-NEXT: [ 602.00 610.00 618.00 ] + +; CHECK-DEMANGLED-INT-INT-INT: Function: _Z3addiii +; CHECK-DEMANGLED-INT-INT-INT-NEXT: [ 921.00 933.00 945.00 ] + +; CHECK-DEMANGLED-DOUBLE-DOUBLE: Function: _Z3adddd +; CHECK-DEMANGLED-DOUBLE-DOUBLE-NEXT: [ 605.00 613.00 621.00 ] + +; CHECK-BB-LEVEL-ALL: Function: _Z3addii +; CHECK-BB-LEVEL-ALL-NEXT: [ 602.00 610.00 618.00 ] +; CHECK-BB-LEVEL-ALL-NEXT: Function: _Z3addiii +; CHECK-BB-LEVEL-ALL-NEXT: [ 921.00 933.00 945.00 ] +; CHECK-BB-LEVEL-ALL-NEXT: Function: _Z3adddd +; CHECK-BB-LEVEL-ALL-NEXT: [ 605.00 613.00 621.00 ] +; CHECK-BB-LEVEL-ALL-NEXT: Function: main +; CHECK-BB-LEVEL-ALL-NEXT: [ 1251.00 1263.00 1275.00 ] + +; CHECK-BB-ADD-INT-INT: Function: _Z3addii +; CHECK-BB-ADD-INT-INT-NEXT: [ 602.00 610.00 618.00 ] + +; CHECK-INST-ADD-INT-INT-INT: Function: _Z3addiii +; CHECK-INST-ADD-INT-INT-INT-NEXT: %4 = alloca i32, align 4 [ 91.00 92.00 93.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %5 = alloca i32, align 4 [ 91.00 92.00 93.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %6 = alloca i32, align 4 [ 91.00 92.00 93.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: store i32 %0, ptr %4, align 4 [ 97.00 98.00 99.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: store i32 %1, ptr %5, align 4 [ 97.00 98.00 99.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: store i32 %2, ptr %6, align 4 [ 97.00 98.00 99.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %7 = load i32, ptr %4, align 4 [ 94.00 95.00 96.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %8 = load i32, ptr %5, align 4 [ 94.00 95.00 96.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %9 = add nsw i32 %7, %8 [ 37.00 38.00 39.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %10 = load i32, ptr %6, align 4 [ 94.00 95.00 96.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: %11 = add nsw i32 %9, %10 [ 37.00 38.00 39.00 ] +; CHECK-INST-ADD-INT-INT-INT-NEXT: ret i32 %11 [ 1.00 2.00 3.00 ] + +; CHECK-NONEXISTENT: error: Function '_Z3subii' not found \ No newline at end of file diff --git a/llvm/tools/llvm-ir2vec/llvm-ir2vec.cpp b/llvm/tools/llvm-ir2vec/llvm-ir2vec.cpp index 7b8d3f093a3d1..ec8a58c3624d8 100644 --- a/llvm/tools/llvm-ir2vec/llvm-ir2vec.cpp +++ b/llvm/tools/llvm-ir2vec/llvm-ir2vec.cpp @@ -55,7 +55,9 @@ //===----------------------------------------------------------------------===// #include "llvm/ADT/ArrayRef.h" +#include "llvm/ADT/STLExtras.h" #include "llvm/Analysis/IR2Vec.h" +#include "llvm/Demangle/Demangle.h" #include "llvm/IR/BasicBlock.h" #include "llvm/IR/Function.h" #include "llvm/IR/Instructions.h" @@ -151,6 +153,26 @@ static cl::opt cl::init(FunctionLevel), cl::sub(EmbeddingsSubCmd), cl::cat(CommonCategory)); +// Helper function to find a function by name +// (supports both exact and demangled) +static const Function *findFunctionByName(const Module &M, + const std::string &FunctionName) { + if (FunctionName.empty()) + return nullptr; + + // Try exact match first + if (auto *ExactMatch = M.getFunction(FunctionName)) + return ExactMatch; + + // Try demangled match + const auto Demangled = llvm::demangle(FunctionName); + auto It = llvm::find_if(M, [&](const Function &Func) { + return llvm::demangle(Func.getName().str()) == Demangled; + }); + + return (It != M.end()) ? &*It : nullptr; +} + namespace ir2vec { /// Relation types for triplet generation @@ -336,7 +358,8 @@ Error processModule(Module &M, raw_ostream &OS) { if (!FunctionName.empty()) { // Process single function - if (const Function *F = M.getFunction(FunctionName)) + const Function *F = findFunctionByName(M, FunctionName); + if (F) Tool.generateEmbeddings(*F, OS); else return createStringError(errc::invalid_argument, @@ -727,7 +750,7 @@ int main(int argc, char **argv) { } else if (EmbeddingsSubCmd) { if (!FunctionName.empty()) { // Process single function - Function *F = M->getFunction(FunctionName); + const Function *F = findFunctionByName(*M, FunctionName); if (!F) { WithColor::error(errs(), ToolName) << "Function '" << FunctionName << "' not found\n";