#define NULL ((void *)0)
typedef unsigned long size_t; // Customize by platform.
typedef long intptr_t;
typedef unsigned long uintptr_t;
typedef long scalar_t__; // Either arithmetic or pointer type.
/* By default, we understand bool (as a convenience). */
typedef int bool;
#define false 0
#define true 1

/* Forward declarations */

/* Type definitions */

/* Variables and functions */
int /*<<< orphan*/ assert(int);
int locate_rank(unsigned n_elems, int n_ranks, unsigned idx) {
  unsigned upp = n_elems / n_ranks, rem = n_elems % n_ranks;
  assert(idx < n_elems);
  if (idx < rem * (upp + 1))
    return idx / (upp + 1);
  else
    return rem + (idx - rem * (upp + 1)) / upp;
}


x86 assembly (clang, -O0):
<locate_rank>:
push   %rbp
mov    %rsp,%rbp
sub    $0x20,%rsp
mov    %edi,-0x8(%rbp)
mov    %esi,-0xc(%rbp)
mov    %edx,-0x10(%rbp)
mov    -0x8(%rbp),%eax
xor    %edx,%edx
divl   -0xc(%rbp)
mov    %eax,-0x14(%rbp)
mov    -0x8(%rbp),%eax
xor    %edx,%edx
divl   -0xc(%rbp)
mov    %edx,-0x18(%rbp)
mov    -0x10(%rbp),%eax
cmp    -0x8(%rbp),%eax
setb   %al
and    $0x1,%al
movzbl %al,%edi
call   1030 <assert@plt>
mov    -0x10(%rbp),%eax
mov    -0x18(%rbp),%ecx
mov    -0x14(%rbp),%edx
add    $0x1,%edx
imul   %edx,%ecx
cmp    %ecx,%eax
jae    1176 <locate_rank+0x66>
mov    -0x10(%rbp),%eax
mov    -0x14(%rbp),%ecx
add    $0x1,%ecx
xor    %edx,%edx
div    %ecx
mov    %eax,-0x4(%rbp)
jmp    119c <locate_rank+0x8c>
mov    -0x18(%rbp),%eax
mov    %eax,-0x1c(%rbp)
mov    -0x10(%rbp),%eax
mov    -0x18(%rbp),%ecx
mov    -0x14(%rbp),%edx
add    $0x1,%edx
imul   %edx,%ecx
sub    %ecx,%eax
xor    %edx,%edx
divl   -0x14(%rbp)
mov    %eax,%ecx
mov    -0x1c(%rbp),%eax
add    %ecx,%eax
mov    %eax,-0x4(%rbp)
mov    -0x4(%rbp),%eax
add    $0x20,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<locate_rank>:
push   %rbp
push   %r14
push   %rbx
mov    %edx,%ebx
mov    %edi,%ecx
mov    %edi,%eax
xor    %edx,%edx
div    %esi
mov    %eax,%ebp
mov    %edx,%r14d
xor    %edi,%edi
cmp    %ecx,%ebx
setb   %dil
call   1030 <assert@plt>
lea    0x1(%rbp),%ecx
mov    %r14d,%eax
imul   %ecx,%eax
cmp    %ebx,%eax
jbe    1145 <locate_rank+0x35>
mov    %ebx,%eax
xor    %edx,%edx
div    %ecx
jmp    1150 <locate_rank+0x40>
sub    %eax,%ebx
mov    %ebx,%eax
xor    %edx,%edx
div    %ebp
add    %r14d,%eax
pop    %rbx
pop    %r14
pop    %rbp
ret

LLVM IR (-O0):
define i32 @locate_rank(i32 %0, i32 %1, i32 %2) {
%4 = alloca i32
%5 = alloca i32
%6 = alloca i32
%7 = alloca i32
%8 = alloca i32
%9 = alloca i32
store i32 %0, i32* %5
store i32 %1, i32* %6
store i32 %2, i32* %7
%10 = load i32, i32* %5
%11 = load i32, i32* %6
%12 = udiv i32 %10, %11
store i32 %12, i32* %8
%13 = load i32, i32* %5
%14 = load i32, i32* %6
%15 = urem i32 %13, %14
store i32 %15, i32* %9
%16 = load i32, i32* %7
%17 = load i32, i32* %5
%18 = icmp ult i32 %16, %17
%19 = zext i1 %18 to i32
%20 = call i32 @assert(i32 %19)
%21 = load i32, i32* %7
%22 = load i32, i32* %9
%23 = load i32, i32* %8
%24 = add i32 %23, 1
%25 = mul i32 %22, %24
%26 = icmp ult i32 %21, %25
br i1 %26, label %27, label %32
27: ; preds = %3
%28 = load i32, i32* %7
%29 = load i32, i32* %8
%30 = add i32 %29, 1
%31 = udiv i32 %28, %30
store i32 %31, i32* %4
br label %43
32: ; preds = %3
%33 = load i32, i32* %9
%34 = load i32, i32* %7
%35 = load i32, i32* %9
%36 = load i32, i32* %8
%37 = add i32 %36, 1
%38 = mul i32 %35, %37
%39 = sub i32 %34, %38
%40 = load i32, i32* %8
%41 = udiv i32 %39, %40
%42 = add i32 %33, %41
store i32 %42, i32* %4
br label %43
43: ; preds = %32, %27
%44 = load i32, i32* %4
ret i32 %44
}

LLVM IR (-O2):
define i32 @locate_rank(i32 %0, i32 %1, i32 %2) {
%4 = udiv i32 %0, %1
%5 = urem i32 %0, %1
%6 = icmp ult i32 %2, %0
%7 = zext i1 %6 to i32
%8 = tail call i32 @assert(i32 %7)
%9 = add i32 %4, 1
%10 = mul i32 %5, %9
%11 = icmp ugt i32 %10, %2
br i1 %11, label %12, label %14
12: ; preds = %3
%13 = udiv i32 %2, %9
br label %18
14: ; preds = %3
%15 = sub i32 %2, %10
%16 = udiv i32 %15, %4
%17 = add i32 %16, %5
br label %18
18: ; preds = %14, %12
%19 = phi i32 [ %13, %12 ], [ %17, %14 ]
ret i32 %19
}