void get_scatter_info(int *sendcnts, int *displs, int n_ranks, int n) {
  int sum = 0;
  int a = n / n_ranks;
  int b = n - a * n_ranks;
  for (int i = 0; i < n_ranks; i++) {
    sendcnts[i] = a + (b > 0 ? 1 : 0);
    displs[i] = sum;
    sum += sendcnts[i];
    b--;
  }
}


x86 assembly (clang, -O0):
<get_scatter_info>:
push   %rbp
mov    %rsp,%rbp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %edx,-0x14(%rbp)
mov    %ecx,-0x18(%rbp)
movl   $0x0,-0x1c(%rbp)
mov    -0x18(%rbp),%eax
cltd
idivl  -0x14(%rbp)
mov    %eax,-0x20(%rbp)
mov    -0x18(%rbp),%eax
mov    -0x20(%rbp),%ecx
imul   -0x14(%rbp),%ecx
sub    %ecx,%eax
mov    %eax,-0x24(%rbp)
movl   $0x0,-0x28(%rbp)
mov    -0x28(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    119b <get_scatter_info+0x9b>
mov    -0x20(%rbp),%edx
mov    -0x24(%rbp),%esi
xor    %eax,%eax
mov    $0x1,%ecx
cmp    $0x0,%esi
cmovg  %ecx,%eax
add    %eax,%edx
mov    -0x8(%rbp),%rax
movslq -0x28(%rbp),%rcx
mov    %edx,(%rax,%rcx,4)
mov    -0x1c(%rbp),%edx
mov    -0x10(%rbp),%rax
movslq -0x28(%rbp),%rcx
mov    %edx,(%rax,%rcx,4)
mov    -0x8(%rbp),%rax
movslq -0x28(%rbp),%rcx
mov    (%rax,%rcx,4),%eax
add    -0x1c(%rbp),%eax
mov    %eax,-0x1c(%rbp)
mov    -0x24(%rbp),%eax
add    $0xffffffff,%eax
mov    %eax,-0x24(%rbp)
mov    -0x28(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x28(%rbp)
jmp    1139 <get_scatter_info+0x39>
pop    %rbp
ret

x86 assembly (clang, -O2):
<get_scatter_info>:
mov    %ecx,%eax
mov    %edx,%ecx
cltd
idiv   %ecx
test   %ecx,%ecx
jle    1184 <get_scatter_info+0x84>
mov    %ecx,%r8d
cmp    $0x1,%ecx
jne    111b <get_scatter_info+0x1b>
xor    %r9d,%r9d
xor    %r11d,%r11d
jmp    116a <get_scatter_info+0x6a>
mov    %r8d,%r10d
and    $0xfffffffe,%r10d
xor    %r9d,%r9d
xor    %r11d,%r11d
nopl   0x0(%rax,%rax,1)
xor    %ecx,%ecx
test   %edx,%edx
setg   %cl
add    %eax,%ecx
mov    %ecx,(%rdi,%r9,4)
mov    %r11d,(%rsi,%r9,4)
add    (%rdi,%r9,4),%r11d
xor    %ecx,%ecx
cmp    $0x2,%edx
setge  %cl
add    %eax,%ecx
mov    %ecx,0x4(%rdi,%r9,4)
mov    %r11d,0x4(%rsi,%r9,4)
add    0x4(%rdi,%r9,4),%r11d
add    $0xfffffffe,%edx
add    $0x2,%r9
cmp    %r9,%r10
jne    1130 <get_scatter_info+0x30>
test   $0x1,%r8b
je     1184 <get_scatter_info+0x84>
xor    %r8d,%r8d
test   %edx,%edx
setg   %r8b
add    %r8d,%eax
mov    %eax,(%rdi,%r9,4)
mov    %r11d,(%rsi,%r9,4)
ret

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

LLVM IR (-O2):
define void @get_scatter_info(i32* nocapture %0, i32* nocapture writeonly %1, i32 %2, i32 %3) {
%5 = sdiv i32 %3, %2
%6 = srem i32 %3, %2
%7 = icmp sgt i32 %2, 0
br i1 %7, label %8, label %26
8: ; preds = %4
%9 = mul nsw i32 %5, %2
%10 = zext i32 %2 to i64
%11 = and i64 %10, 1
%12 = icmp eq i32 %2, 1
br i1 %12, label %15, label %13
13: ; preds = %8
%14 = and i64 %10, 4294967294
br label %27
15: ; preds = %27, %8
%16 = phi i64 [ 0, %8 ], [ %48, %27 ]
%17 = phi i32 [ %6, %8 ], [ %47, %27 ]
%18 = phi i32 [ 0, %8 ], [ %46, %27 ]
%19 = icmp eq i64 %11, 0
br i1 %19, label %26, label %20
20: ; preds = %15
%21 = icmp sgt i32 %17, 0
%22 = zext i1 %21 to i32
%23 = add nsw i32 %5, %22
%24 = getelementptr inbounds i32, i32* %0, i64 %16
store i32 %23, i32* %24
%25 = getelementptr inbounds i32, i32* %1, i64 %16
store i32 %18, i32* %25
br label %26
26: ; preds = %20, %15, %4
ret void
27: ; preds = %27, %13
%28 = phi i64 [ 0, %13 ], [ %48, %27 ]
%29 = phi i32 [ %6, %13 ], [ %47, %27 ]
%30 = phi i32 [ 0, %13 ], [ %46, %27 ]
%31 = phi i64 [ 0, %13 ], [ %49, %27 ]
%32 = icmp sgt i32 %29, 0
%33 = zext i1 %32 to i32
%34 = add nsw i32 %5, %33
%35 = getelementptr inbounds i32, i32* %0, i64 %28
store i32 %34, i32* %35
%36 = getelementptr inbounds i32, i32* %1, i64 %28
store i32 %30, i32* %36
%37 = load i32, i32* %35
%38 = add nsw i32 %37, %30
%39 = or i64 %28, 1
%40 = icmp sgt i32 %29, 1
%41 = zext i1 %40 to i32
%42 = add nsw i32 %5, %41
%43 = getelementptr inbounds i32, i32* %0, i64 %39
store i32 %42, i32* %43
%44 = getelementptr inbounds i32, i32* %1, i64 %39
store i32 %38, i32* %44
%45 = load i32, i32* %43
%46 = add nsw i32 %45, %38
%47 = add nsw i32 %29, -2
%48 = add nuw nsw i64 %28, 2
%49 = add i64 %31, 2
%50 = icmp eq i64 %49, %14
br i1 %50, label %15, label %27
}