#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 */
void *calloc(int, int);
float **one_hot_encode(float *a, int n, int k) {
  int i;
  float **t = calloc(n, sizeof(float *));
  for (i = 0; i < n; ++i) {
    t[i] = calloc(k, sizeof(float));
    int index = (int)a[i];
    t[i][index] = 1;
  }
  return t;
}


x86 assembly (clang, -O0):
<one_hot_encode>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %rdi,-0x8(%rbp)
mov    %esi,-0xc(%rbp)
mov    %edx,-0x10(%rbp)
mov    -0xc(%rbp),%edi
mov    $0x8,%esi
call   1030 <calloc@plt>
mov    %rax,-0x20(%rbp)
movl   $0x0,-0x14(%rbp)
mov    -0x14(%rbp),%eax
cmp    -0xc(%rbp),%eax
jge    119d <one_hot_encode+0x8d>
mov    -0x10(%rbp),%edi
mov    $0x4,%esi
call   1030 <calloc@plt>
mov    %rax,%rdx
mov    -0x20(%rbp),%rax
movslq -0x14(%rbp),%rcx
mov    %rdx,(%rax,%rcx,8)
mov    -0x8(%rbp),%rax
movslq -0x14(%rbp),%rcx
cvttss2si (%rax,%rcx,4),%eax
mov    %eax,-0x24(%rbp)
mov    -0x20(%rbp),%rax
movslq -0x14(%rbp),%rcx
mov    (%rax,%rcx,8),%rax
movslq -0x24(%rbp),%rcx
movss  0xe76(%rip),%xmm0
movss  %xmm0,(%rax,%rcx,4)
mov    -0x14(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x14(%rbp)
jmp    113a <one_hot_encode+0x2a>
mov    -0x20(%rbp),%rax
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<one_hot_encode>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
push   %rax
mov    %edx,%r14d
mov    %esi,%r15d
mov    %rdi,%r12
mov    %esi,%edi
mov    $0x8,%esi
call   1030 <calloc@plt>
mov    %rax,%r13
test   %r15d,%r15d
jle    116b <one_hot_encode+0x5b>
mov    %r15d,%ebx
xor    %ebp,%ebp
nopl   (%rax)
mov    %r14d,%edi
mov    $0x4,%esi
call   1030 <calloc@plt>
mov    %rax,0x0(%r13,%rbp,8)
cvttss2si (%r12,%rbp,4),%ecx
movslq %ecx,%rcx
movl   $0x3f800000,(%rax,%rcx,4)
add    $0x1,%rbp
cmp    %rbp,%rbx
jne    1140 <one_hot_encode+0x30>
mov    %r13,%rax
add    $0x8,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
define float** @one_hot_encode(float* %0, i32 %1, i32 %2) {
%4 = alloca float*
%5 = alloca i32
%6 = alloca i32
%7 = alloca i32
%8 = alloca float**
%9 = alloca i32
store float* %0, float** %4
store i32 %1, i32* %5
store i32 %2, i32* %6
%10 = load i32, i32* %5
%11 = call i8* @calloc(i32 %10, i32 8)
%12 = bitcast i8* %11 to float**
store float** %12, float*** %8
store i32 0, i32* %7
br label %13
13: ; preds = %39, %3
%14 = load i32, i32* %7
%15 = load i32, i32* %5
%16 = icmp slt i32 %14, %15
br i1 %16, label %17, label %42
17: ; preds = %13
%18 = load i32, i32* %6
%19 = call i8* @calloc(i32 %18, i32 4)
%20 = bitcast i8* %19 to float*
%21 = load float**, float*** %8
%22 = load i32, i32* %7
%23 = sext i32 %22 to i64
%24 = getelementptr inbounds float*, float** %21, i64 %23
store float* %20, float** %24
%25 = load float*, float** %4
%26 = load i32, i32* %7
%27 = sext i32 %26 to i64
%28 = getelementptr inbounds float, float* %25, i64 %27
%29 = load float, float* %28
%30 = fptosi float %29 to i32
store i32 %30, i32* %9
%31 = load float**, float*** %8
%32 = load i32, i32* %7
%33 = sext i32 %32 to i64
%34 = getelementptr inbounds float*, float** %31, i64 %33
%35 = load float*, float** %34
%36 = load i32, i32* %9
%37 = sext i32 %36 to i64
%38 = getelementptr inbounds float, float* %35, i64 %37
store float 1.000000e+00, float* %38
br label %39
39: ; preds = %17
%40 = load i32, i32* %7
%41 = add nsw i32 %40, 1
store i32 %41, i32* %7
br label %13
42: ; preds = %13
%43 = load float**, float*** %8
ret float** %43
}

LLVM IR (-O2):
define noalias float** @one_hot_encode(float* nocapture %0, i32 %1, i32 %2) {
%4 = tail call i8* @calloc(i32 %1, i32 8)
%5 = bitcast i8* %4 to float**
%6 = icmp sgt i32 %1, 0
br i1 %6, label %7, label %22
7: ; preds = %3
%8 = zext i32 %1 to i64
br label %9
9: ; preds = %7, %9
%10 = phi i64 [ 0, %7 ], [ %20, %9 ]
%11 = tail call i8* @calloc(i32 %2, i32 4)
%12 = bitcast i8* %11 to float*
%13 = getelementptr inbounds float*, float** %5, i64 %10
%14 = bitcast float** %13 to i8**
store i8* %11, i8** %14
%15 = getelementptr inbounds float, float* %0, i64 %10
%16 = load float, float* %15
%17 = fptosi float %16 to i32
%18 = sext i32 %17 to i64
%19 = getelementptr inbounds float, float* %12, i64 %18
store float 1.000000e+00, float* %19
%20 = add nuw nsw i64 %10, 1
%21 = icmp eq i64 %20, %8
br i1 %21, label %22, label %9
22: ; preds = %9, %3
ret float** %5
}