void init_lens(long *lens, int *sent_lens, int batch_size, int batch) {
  lens[0] = sent_lens[batch * batch_size];
  for (int i = 1; i < batch_size; i++) {
    lens[i] = lens[i - 1] + sent_lens[batch * batch_size + i];
  }
}


x86 assembly (clang, -O0):
<init_lens>:
push   %rbp
mov    %rsp,%rbp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %edx,-0x14(%rbp)
mov    %ecx,-0x18(%rbp)
mov    -0x10(%rbp),%rax
mov    -0x18(%rbp),%ecx
imul   -0x14(%rbp),%ecx
movslq %ecx,%rcx
movslq (%rax,%rcx,4),%rcx
mov    -0x8(%rbp),%rax
mov    %rcx,(%rax)
movl   $0x1,-0x1c(%rbp)
mov    -0x1c(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    1181 <init_lens+0x81>
mov    -0x8(%rbp),%rax
mov    -0x1c(%rbp),%ecx
sub    $0x1,%ecx
movslq %ecx,%rcx
mov    (%rax,%rcx,8),%rdx
mov    -0x10(%rbp),%rax
mov    -0x18(%rbp),%ecx
imul   -0x14(%rbp),%ecx
add    -0x1c(%rbp),%ecx
movslq %ecx,%rcx
movslq (%rax,%rcx,4),%rax
add    %rax,%rdx
mov    -0x8(%rbp),%rax
movslq -0x1c(%rbp),%rcx
mov    %rdx,(%rax,%rcx,8)
mov    -0x1c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x1c(%rbp)
jmp    1132 <init_lens+0x32>
pop    %rbp
ret

x86 assembly (clang, -O2):
<init_lens>:
imul   %edx,%ecx
movslq %ecx,%r8
movslq (%rsi,%r8,4),%rax
mov    %rax,(%rdi)
cmp    $0x2,%edx
jl     11b4 <init_lens+0xb4>
mov    %edx,%ecx
lea    -0x1(%rcx),%r10
add    $0xfffffffffffffffe,%rcx
mov    %r10d,%r9d
and    $0x3,%r9d
mov    $0x1,%edx
cmp    $0x3,%rcx
jb     1180 <init_lens+0x80>
and    $0xfffffffffffffffc,%r10
lea    (%rsi,%r8,4),%r11
add    $0x10,%r11
xor    %edx,%edx
movslq -0xc(%r11,%rdx,4),%rcx
add    %rax,%rcx
mov    %rcx,0x8(%rdi,%rdx,8)
movslq -0x8(%r11,%rdx,4),%rax
add    %rcx,%rax
mov    %rax,0x10(%rdi,%rdx,8)
movslq -0x4(%r11,%rdx,4),%rcx
add    %rax,%rcx
mov    %rcx,0x18(%rdi,%rdx,8)
movslq (%r11,%rdx,4),%rax
add    %rcx,%rax
mov    %rax,0x20(%rdi,%rdx,8)
add    $0x4,%rdx
cmp    %rdx,%r10
jne    1140 <init_lens+0x40>
add    $0x1,%rdx
test   %r9,%r9
je     11b4 <init_lens+0xb4>
lea    (%rdi,%rdx,8),%rcx
add    %rdx,%r8
lea    (%rsi,%r8,4),%rdx
xor    %esi,%esi
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax)
movslq (%rdx,%rsi,4),%rdi
add    %rdi,%rax
mov    %rax,(%rcx,%rsi,8)
add    $0x1,%rsi
cmp    %rsi,%r9
jne    11a0 <init_lens+0xa0>
ret

LLVM IR (-O0):
define void @init_lens(i64* %0, i32* %1, i32 %2, i32 %3) {
%5 = alloca i64*
%6 = alloca i32*
%7 = alloca i32
%8 = alloca i32
%9 = alloca i32
store i64* %0, i64** %5
store i32* %1, i32** %6
store i32 %2, i32* %7
store i32 %3, i32* %8
%10 = load i32*, i32** %6
%11 = load i32, i32* %8
%12 = load i32, i32* %7
%13 = mul nsw i32 %11, %12
%14 = sext i32 %13 to i64
%15 = getelementptr inbounds i32, i32* %10, i64 %14
%16 = load i32, i32* %15
%17 = sext i32 %16 to i64
%18 = load i64*, i64** %5
%19 = getelementptr inbounds i64, i64* %18, i64 0
store i64 %17, i64* %19
store i32 1, i32* %9
br label %20
20: ; preds = %46, %4
%21 = load i32, i32* %9
%22 = load i32, i32* %7
%23 = icmp slt i32 %21, %22
br i1 %23, label %24, label %49
24: ; preds = %20
%25 = load i64*, i64** %5
%26 = load i32, i32* %9
%27 = sub nsw i32 %26, 1
%28 = sext i32 %27 to i64
%29 = getelementptr inbounds i64, i64* %25, i64 %28
%30 = load i64, i64* %29
%31 = load i32*, i32** %6
%32 = load i32, i32* %8
%33 = load i32, i32* %7
%34 = mul nsw i32 %32, %33
%35 = load i32, i32* %9
%36 = add nsw i32 %34, %35
%37 = sext i32 %36 to i64
%38 = getelementptr inbounds i32, i32* %31, i64 %37
%39 = load i32, i32* %38
%40 = sext i32 %39 to i64
%41 = add nsw i64 %30, %40
%42 = load i64*, i64** %5
%43 = load i32, i32* %9
%44 = sext i32 %43 to i64
%45 = getelementptr inbounds i64, i64* %42, i64 %44
store i64 %41, i64* %45
br label %46
46: ; preds = %24
%47 = load i32, i32* %9
%48 = add nsw i32 %47, 1
store i32 %48, i32* %9
br label %20
49: ; preds = %20
ret void
}

LLVM IR (-O2):
define void @init_lens(i64* nocapture %0, i32* nocapture %1, i32 %2, i32 %3) {
%5 = mul nsw i32 %3, %2
%6 = sext i32 %5 to i64
%7 = getelementptr inbounds i32, i32* %1, i64 %6
%8 = load i32, i32* %7
%9 = sext i32 %8 to i64
store i64 %9, i64* %0
%10 = icmp sgt i32 %2, 1
br i1 %10, label %11, label %36
11: ; preds = %4
%12 = zext i32 %2 to i64
%13 = add nsw i64 %12, -1
%14 = add nsw i64 %12, -2
%15 = and i64 %13, 3
%16 = icmp ult i64 %14, 3
br i1 %16, label %19, label %17
17: ; preds = %11
%18 = and i64 %13, -4
br label %37
19: ; preds = %37, %11
%20 = phi i64 [ %9, %11 ], [ %66, %37 ]
%21 = phi i64 [ 1, %11 ], [ %68, %37 ]
%22 = icmp eq i64 %15, 0
br i1 %22, label %36, label %23
23: ; preds = %19, %23
%24 = phi i64 [ %31, %23 ], [ %20, %19 ]
%25 = phi i64 [ %33, %23 ], [ %21, %19 ]
%26 = phi i64 [ %34, %23 ], [ 0, %19 ]
%27 = add nsw i64 %25, %6
%28 = getelementptr inbounds i32, i32* %1, i64 %27
%29 = load i32, i32* %28
%30 = sext i32 %29 to i64
%31 = add nsw i64 %24, %30
%32 = getelementptr inbounds i64, i64* %0, i64 %25
store i64 %31, i64* %32
%33 = add nuw nsw i64 %25, 1
%34 = add i64 %26, 1
%35 = icmp eq i64 %34, %15
br i1 %35, label %36, label %23
36: ; preds = %19, %23, %4
ret void
37: ; preds = %37, %17
%38 = phi i64 [ %9, %17 ], [ %66, %37 ]
%39 = phi i64 [ 1, %17 ], [ %68, %37 ]
%40 = phi i64 [ 0, %17 ], [ %69, %37 ]
%41 = add nsw i64 %39, %6
%42 = getelementptr inbounds i32, i32* %1, i64 %41
%43 = load i32, i32* %42
%44 = sext i32 %43 to i64
%45 = add nsw i64 %38, %44
%46 = getelementptr inbounds i64, i64* %0, i64 %39
store i64 %45, i64* %46
%47 = add nuw nsw i64 %39, 1
%48 = add nsw i64 %47, %6
%49 = getelementptr inbounds i32, i32* %1, i64 %48
%50 = load i32, i32* %49
%51 = sext i32 %50 to i64
%52 = add nsw i64 %45, %51
%53 = getelementptr inbounds i64, i64* %0, i64 %47
store i64 %52, i64* %53
%54 = add nuw nsw i64 %39, 2
%55 = add nsw i64 %54, %6
%56 = getelementptr inbounds i32, i32* %1, i64 %55
%57 = load i32, i32* %56
%58 = sext i32 %57 to i64
%59 = add nsw i64 %52, %58
%60 = getelementptr inbounds i64, i64* %0, i64 %54
store i64 %59, i64* %60
%61 = add nuw nsw i64 %39, 3
%62 = add nsw i64 %61, %6
%63 = getelementptr inbounds i32, i32* %1, i64 %62
%64 = load i32, i32* %63
%65 = sext i32 %64 to i64
%66 = add nsw i64 %59, %65
%67 = getelementptr inbounds i64, i64* %0, i64 %61
store i64 %66, i64* %67
%68 = add nuw nsw i64 %39, 4
%69 = add i64 %40, 4
%70 = icmp eq i64 %69, %18
br i1 %70, label %19, label %37
}