#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 */
typedef scalar_t__ uint32_t;
typedef scalar_t__ int32_t;

/* Variables and functions */
scalar_t__ partition_basic(float *, scalar_t__, scalar_t__, int);
void find_top_k_basic(float *input, uint32_t input_len, uint32_t k) {
  int32_t low = 0;
  int32_t high = input_len - 1;
  int32_t j = partition_basic(input, low, high, 0);
  while (j != k) {
    if (k > j) {
      low = j + 1;
    } else {
      high = j;
    }
    j = partition_basic(input, low, high, 0);
  }
  partition_basic(input, 0, k - 1, 1);
}


x86 assembly (clang, -O0):
<find_top_k_basic>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %rdx,-0x18(%rbp)
movq   $0x0,-0x20(%rbp)
mov    -0x10(%rbp),%rax
sub    $0x1,%rax
mov    %rax,-0x28(%rbp)
mov    -0x8(%rbp),%rdi
mov    -0x20(%rbp),%rsi
mov    -0x28(%rbp),%rdx
xor    %ecx,%ecx
call   1030 <partition_basic@plt>
mov    %rax,-0x30(%rbp)
mov    -0x30(%rbp),%rax
cmp    -0x18(%rbp),%rax
je     11a0 <find_top_k_basic+0x90>
mov    -0x18(%rbp),%rax
cmp    -0x30(%rbp),%rax
jle    117c <find_top_k_basic+0x6c>
mov    -0x30(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x20(%rbp)
jmp    1184 <find_top_k_basic+0x74>
mov    -0x30(%rbp),%rax
mov    %rax,-0x28(%rbp)
mov    -0x8(%rbp),%rdi
mov    -0x20(%rbp),%rsi
mov    -0x28(%rbp),%rdx
xor    %ecx,%ecx
call   1030 <partition_basic@plt>
mov    %rax,-0x30(%rbp)
jmp    114f <find_top_k_basic+0x3f>
mov    -0x8(%rbp),%rdi
mov    -0x18(%rbp),%rdx
sub    $0x1,%rdx
xor    %eax,%eax
mov    %eax,%esi
mov    $0x1,%ecx
call   1030 <partition_basic@plt>
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<find_top_k_basic>:
push   %r15
push   %r14
push   %r12
push   %rbx
push   %rax
mov    %rdx,%r15
mov    %rsi,%r12
mov    %rdi,%r14
add    $0xffffffffffffffff,%r12
xor    %esi,%esi
mov    %r12,%rdx
xor    %ecx,%ecx
call   1030 <partition_basic@plt>
cmp    %r15,%rax
je     1164 <find_top_k_basic+0x54>
xor    %ebx,%ebx
nopl   0x0(%rax,%rax,1)
lea    0x1(%rax),%rcx
cmp    %r15,%rax
cmovl  %rcx,%rbx
cmovge %rax,%r12
mov    %r14,%rdi
mov    %rbx,%rsi
mov    %r12,%rdx
xor    %ecx,%ecx
call   1030 <partition_basic@plt>
cmp    %r15,%rax
jne    1140 <find_top_k_basic+0x30>
add    $0xffffffffffffffff,%r15
mov    %r14,%rdi
xor    %esi,%esi
mov    %r15,%rdx
mov    $0x1,%ecx
add    $0x8,%rsp
pop    %rbx
pop    %r12
pop    %r14
pop    %r15
jmp    1030 <partition_basic@plt>

LLVM IR (-O0):
define void @find_top_k_basic(float* %0, i64 %1, i64 %2) {
%4 = alloca float*
%5 = alloca i64
%6 = alloca i64
%7 = alloca i64
%8 = alloca i64
%9 = alloca i64
store float* %0, float** %4
store i64 %1, i64* %5
store i64 %2, i64* %6
store i64 0, i64* %7
%10 = load i64, i64* %5
%11 = sub nsw i64 %10, 1
store i64 %11, i64* %8
%12 = load float*, float** %4
%13 = load i64, i64* %7
%14 = load i64, i64* %8
%15 = call i64 @partition_basic(float* %12, i64 %13, i64 %14, i32 0)
store i64 %15, i64* %9
br label %16
16: ; preds = %29, %3
%17 = load i64, i64* %9
%18 = load i64, i64* %6
%19 = icmp ne i64 %17, %18
br i1 %19, label %20, label %34
20: ; preds = %16
%21 = load i64, i64* %6
%22 = load i64, i64* %9
%23 = icmp sgt i64 %21, %22
br i1 %23, label %24, label %27
24: ; preds = %20
%25 = load i64, i64* %9
%26 = add nsw i64 %25, 1
store i64 %26, i64* %7
br label %29
27: ; preds = %20
%28 = load i64, i64* %9
store i64 %28, i64* %8
br label %29
29: ; preds = %27, %24
%30 = load float*, float** %4
%31 = load i64, i64* %7
%32 = load i64, i64* %8
%33 = call i64 @partition_basic(float* %30, i64 %31, i64 %32, i32 0)
store i64 %33, i64* %9
br label %16
34: ; preds = %16
%35 = load float*, float** %4
%36 = load i64, i64* %6
%37 = sub nsw i64 %36, 1
%38 = call i64 @partition_basic(float* %35, i64 0, i64 %37, i32 1)
ret void
}

LLVM IR (-O2):
define void @find_top_k_basic(float* %0, i64 %1, i64 %2) {
%4 = add nsw i64 %1, -1
%5 = tail call i64 @partition_basic(float* %0, i64 0, i64 %4, i32 0)
%6 = icmp eq i64 %5, %2
br i1 %6, label %17, label %7
7: ; preds = %3, %7
%8 = phi i64 [ %15, %7 ], [ %5, %3 ]
%9 = phi i64 [ %14, %7 ], [ %4, %3 ]
%10 = phi i64 [ %13, %7 ], [ 0, %3 ]
%11 = icmp slt i64 %8, %2
%12 = add nsw i64 %8, 1
%13 = select i1 %11, i64 %12, i64 %10
%14 = select i1 %11, i64 %9, i64 %8
%15 = tail call i64 @partition_basic(float* %0, i64 %13, i64 %14, i32 0)
%16 = icmp eq i64 %15, %2
br i1 %16, label %17, label %7
17: ; preds = %7, %3
%18 = add nsw i64 %2, -1
%19 = tail call i64 @partition_basic(float* %0, i64 0, i64 %18, i32 1)
ret void
}