e86bd05bdc
This PR restructures the vocabulary. * String based look-ups are removed. Vocabulary is changed from a map to vector. (#141832) * Grouped all the vocabulary related methods under a single class - `ir2vec::Vocabulary`. This replaces `IR2VecVocabResult`. * `ir2vec::Vocabulary` effectively abstracts out the _layout_ and other internal details of the vector structure. Exposes necessary APIs for accessing the Vocabulary. These changes ensure that _all_ known opcodes and types are present in the vocabulary. We have retained the original operands. This can be extended going forward. (Tracking issue - #141817)
39 lines
1.3 KiB
LLVM
39 lines
1.3 KiB
LLVM
; RUN: opt -passes='print<ir2vec>' -o /dev/null -ir2vec-vocab-path=%S/Inputs/dummy_3D_nonzero_opc_vocab.json %s 2>&1 | FileCheck %s
|
|
|
|
define dso_local i32 @abc(i32 noundef %a, i32 noundef %b) #0 {
|
|
entry:
|
|
%retval = alloca i32, align 4
|
|
%a.addr = alloca i32, align 4
|
|
%b.addr = alloca i32, align 4
|
|
store i32 %a, ptr %a.addr, align 4
|
|
store i32 %b, ptr %b.addr, align 4
|
|
%0 = load i32, ptr %a.addr, align 4
|
|
%1 = load i32, ptr %b.addr, align 4
|
|
%cmp = icmp sgt i32 %0, %1
|
|
br i1 %cmp, label %if.then, label %if.else
|
|
|
|
if.then: ; preds = %entry
|
|
%2 = load i32, ptr %b.addr, align 4
|
|
store i32 %2, ptr %retval, align 4
|
|
br label %return
|
|
|
|
if.else: ; preds = %entry
|
|
%3 = load i32, ptr %a.addr, align 4
|
|
store i32 %3, ptr %retval, align 4
|
|
br label %return
|
|
|
|
return: ; preds = %if.else, %if.then
|
|
%4 = load i32, ptr %retval, align 4
|
|
ret i32 %4
|
|
}
|
|
|
|
; CHECK: Basic block vectors:
|
|
; CHECK-NEXT: Basic block: entry:
|
|
; CHECK-NEXT: [ 816.00 825.00 834.00 ]
|
|
; CHECK-NEXT: Basic block: if.then:
|
|
; CHECK-NEXT: [ 195.00 198.00 201.00 ]
|
|
; CHECK-NEXT: Basic block: if.else:
|
|
; CHECK-NEXT: [ 195.00 198.00 201.00 ]
|
|
; CHECK-NEXT: Basic block: return:
|
|
; CHECK-NEXT: [ 95.00 97.00 99.00 ]
|