#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*/ free(size_t *);
int kpt_relocate_dense_BZ_grid_address(int **, size_t *, int **, int const *,
                                       double **, int const *);
scalar_t__ malloc(int);
int kpt_relocate_BZ_grid_address(int bz_grid_address[][3], int bz_map[],
                                 int grid_address[][3], const int mesh[3],
                                 double rec_lattice[3][3],
                                 const int is_shift[3]) {
  int i, num_bz_map, num_bzgp;
  size_t *dense_bz_map;
  num_bz_map = mesh[0] * mesh[1] * mesh[2] * 8;
  if ((dense_bz_map = (size_t *)malloc(sizeof(size_t) * num_bz_map)) == NULL) {
    ;
    return 0;
  }
  num_bzgp = kpt_relocate_dense_BZ_grid_address(
      bz_grid_address, dense_bz_map, grid_address, mesh, rec_lattice, is_shift);
  for (i = 0; i < num_bz_map; i++) {
    if (dense_bz_map[i] == num_bz_map) {
      bz_map[i] = -1;
    } else {
      bz_map[i] = dense_bz_map[i];
    }
  }
  free(dense_bz_map);
  dense_bz_map = NULL;
  return num_bzgp;
}


x86 assembly (clang, -O0):
<kpt_relocate_BZ_grid_address>:
push   %rbp
mov    %rsp,%rbp
sub    $0x50,%rsp
mov    %rdi,-0x10(%rbp)
mov    %rsi,-0x18(%rbp)
mov    %rdx,-0x20(%rbp)
mov    %rcx,-0x28(%rbp)
mov    %r8,-0x30(%rbp)
mov    %r9,-0x38(%rbp)
mov    -0x28(%rbp),%rax
mov    (%rax),%eax
mov    -0x28(%rbp),%rcx
imul   0x4(%rcx),%eax
mov    -0x28(%rbp),%rcx
imul   0x8(%rcx),%eax
shl    $0x3,%eax
mov    %eax,-0x40(%rbp)
movslq -0x40(%rbp),%rax
shl    $0x3,%rax
mov    %eax,%edi
call   1050 <malloc@plt>
mov    %rax,-0x50(%rbp)
cmp    $0x0,%rax
jne    1195 <kpt_relocate_BZ_grid_address+0x65>
movl   $0x0,-0x4(%rbp)
jmp    1238 <kpt_relocate_BZ_grid_address+0x108>
mov    -0x10(%rbp),%rdi
mov    -0x50(%rbp),%rsi
mov    -0x20(%rbp),%rdx
mov    -0x28(%rbp),%rcx
mov    -0x30(%rbp),%r8
mov    -0x38(%rbp),%r9
call   1040 <kpt_relocate_dense_BZ_grid_address@plt>
mov    %eax,-0x44(%rbp)
movl   $0x0,-0x3c(%rbp)
mov    -0x3c(%rbp),%eax
cmp    -0x40(%rbp),%eax
jge    1221 <kpt_relocate_BZ_grid_address+0xf1>
mov    -0x50(%rbp),%rax
movslq -0x3c(%rbp),%rcx
mov    (%rax,%rcx,8),%rax
movslq -0x40(%rbp),%rcx
cmp    %rcx,%rax
jne    11f5 <kpt_relocate_BZ_grid_address+0xc5>
mov    -0x18(%rbp),%rax
movslq -0x3c(%rbp),%rcx
movl   $0xffffffff,(%rax,%rcx,4)
jmp    120e <kpt_relocate_BZ_grid_address+0xde>
mov    -0x50(%rbp),%rax
movslq -0x3c(%rbp),%rcx
mov    (%rax,%rcx,8),%rax
mov    %eax,%edx
mov    -0x18(%rbp),%rax
movslq -0x3c(%rbp),%rcx
mov    %edx,(%rax,%rcx,4)
jmp    1213 <kpt_relocate_BZ_grid_address+0xe3>
mov    -0x3c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x3c(%rbp)
jmp    11bc <kpt_relocate_BZ_grid_address+0x8c>
mov    -0x50(%rbp),%rdi
call   1030 <free@plt>
movq   $0x0,-0x50(%rbp)
mov    -0x44(%rbp),%eax
mov    %eax,-0x4(%rbp)
mov    -0x4(%rbp),%eax
add    $0x50,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<kpt_relocate_BZ_grid_address>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
sub    $0x18,%rsp
mov    %rcx,%r14
mov    0x4(%rcx),%r12d
imul   (%rcx),%r12d
mov    %r9,0x10(%rsp)
mov    %r8,0x8(%rsp)
mov    %rdx,%r13
mov    %rsi,%rbp
mov    %rdi,%r15
imul   0x8(%rcx),%r12d
mov    %r12d,%edi
shl    $0x6,%edi
call   1050 <malloc@plt>
test   %rax,%rax
je     11c2 <kpt_relocate_BZ_grid_address+0x92>
mov    %rax,%rbx
mov    %r15,%rdi
mov    %rax,%rsi
mov    %r13,%rdx
mov    %r14,%rcx
mov    0x8(%rsp),%r8
mov    0x10(%rsp),%r9
call   1040 <kpt_relocate_dense_BZ_grid_address@plt>
mov    %eax,%r14d
test   %r12d,%r12d
jle    1256 <kpt_relocate_BZ_grid_address+0x126>
shl    $0x3,%r12d
movslq %r12d,%rax
cmp    $0x2,%eax
mov    $0x1,%edx
cmovge %eax,%edx
lea    -0x1(%rdx),%rsi
mov    %edx,%r8d
and    $0x1,%r8d
cmp    $0x3,%rsi
jae    11ca <kpt_relocate_BZ_grid_address+0x9a>
xor    %esi,%esi
jmp    1224 <kpt_relocate_BZ_grid_address+0xf4>
xor    %r14d,%r14d
jmp    125e <kpt_relocate_BZ_grid_address+0x12e>
and    $0x7ffffff8,%edx
xor    %esi,%esi
mov    $0xffffffff,%edi
nopw   0x0(%rax,%rax,1)
mov    (%rbx,%rsi,8),%rcx
cmp    %rax,%rcx
cmove  %edi,%ecx
mov    %ecx,0x0(%rbp,%rsi,4)
mov    0x8(%rbx,%rsi,8),%rcx
cmp    %rax,%rcx
cmove  %edi,%ecx
mov    %ecx,0x4(%rbp,%rsi,4)
mov    0x10(%rbx,%rsi,8),%rcx
cmp    %rax,%rcx
cmove  %edi,%ecx
mov    %ecx,0x8(%rbp,%rsi,4)
mov    0x18(%rbx,%rsi,8),%rcx
cmp    %rax,%rcx
cmove  %edi,%ecx
mov    %ecx,0xc(%rbp,%rsi,4)
add    $0x4,%rsi
cmp    %rsi,%rdx
jne    11e0 <kpt_relocate_BZ_grid_address+0xb0>
test   %r8,%r8
je     1256 <kpt_relocate_BZ_grid_address+0x126>
lea    0x0(,%rsi,4),%rdx
add    %rbp,%rdx
lea    (%rbx,%rsi,8),%rsi
xor    %edi,%edi
mov    $0xffffffff,%ebp
nop
mov    (%rsi,%rdi,8),%rcx
cmp    %rax,%rcx
cmove  %ebp,%ecx
mov    %ecx,(%rdx,%rdi,4)
add    $0x1,%rdi
cmp    %rdi,%r8
jne    1240 <kpt_relocate_BZ_grid_address+0x110>
mov    %rbx,%rdi
call   1030 <free@plt>
mov    %r14d,%eax
add    $0x18,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
define i32 @kpt_relocate_BZ_grid_address([3 x i32]* %0, i32* %1, [3 x i32]* %2, i32* %3, [3 x double]* %4, i32* %5) {
%7 = alloca i32
%8 = alloca [3 x i32]*
%9 = alloca i32*
%10 = alloca [3 x i32]*
%11 = alloca i32*
%12 = alloca [3 x double]*
%13 = alloca i32*
%14 = alloca i32
%15 = alloca i32
%16 = alloca i32
%17 = alloca i64*
store [3 x i32]* %0, [3 x i32]** %8
store i32* %1, i32** %9
store [3 x i32]* %2, [3 x i32]** %10
store i32* %3, i32** %11
store [3 x double]* %4, [3 x double]** %12
store i32* %5, i32** %13
%18 = load i32*, i32** %11
%19 = getelementptr inbounds i32, i32* %18, i64 0
%20 = load i32, i32* %19
%21 = load i32*, i32** %11
%22 = getelementptr inbounds i32, i32* %21, i64 1
%23 = load i32, i32* %22
%24 = mul nsw i32 %20, %23
%25 = load i32*, i32** %11
%26 = getelementptr inbounds i32, i32* %25, i64 2
%27 = load i32, i32* %26
%28 = mul nsw i32 %24, %27
%29 = mul nsw i32 %28, 8
store i32 %29, i32* %15
%30 = load i32, i32* %15
%31 = sext i32 %30 to i64
%32 = mul i64 8, %31
%33 = trunc i64 %32 to i32
%34 = call i64 @malloc(i32 %33)
%35 = inttoptr i64 %34 to i64*
store i64* %35, i64** %17
%36 = icmp eq i64* %35, null
br i1 %36, label %37, label %38
37: ; preds = %6
store i32 0, i32* %7
br label %86
38: ; preds = %6
%39 = load [3 x i32]*, [3 x i32]** %8
%40 = bitcast [3 x i32]* %39 to i32**
%41 = load i64*, i64** %17
%42 = load [3 x i32]*, [3 x i32]** %10
%43 = bitcast [3 x i32]* %42 to i32**
%44 = load i32*, i32** %11
%45 = load [3 x double]*, [3 x double]** %12
%46 = bitcast [3 x double]* %45 to double**
%47 = load i32*, i32** %13
%48 = call i32 @kpt_relocate_dense_BZ_grid_address(i32** %40, i64* %41, i32** %43, i32* %44, double** %46, i32* %47)
store i32 %48, i32* %16
store i32 0, i32* %14
br label %49
49: ; preds = %79, %38
%50 = load i32, i32* %14
%51 = load i32, i32* %15
%52 = icmp slt i32 %50, %51
br i1 %52, label %53, label %82
53: ; preds = %49
%54 = load i64*, i64** %17
%55 = load i32, i32* %14
%56 = sext i32 %55 to i64
%57 = getelementptr inbounds i64, i64* %54, i64 %56
%58 = load i64, i64* %57
%59 = load i32, i32* %15
%60 = sext i32 %59 to i64
%61 = icmp eq i64 %58, %60
br i1 %61, label %62, label %67
62: ; preds = %53
%63 = load i32*, i32** %9
%64 = load i32, i32* %14
%65 = sext i32 %64 to i64
%66 = getelementptr inbounds i32, i32* %63, i64 %65
store i32 -1, i32* %66
br label %78
67: ; preds = %53
%68 = load i64*, i64** %17
%69 = load i32, i32* %14
%70 = sext i32 %69 to i64
%71 = getelementptr inbounds i64, i64* %68, i64 %70
%72 = load i64, i64* %71
%73 = trunc i64 %72 to i32
%74 = load i32*, i32** %9
%75 = load i32, i32* %14
%76 = sext i32 %75 to i64
%77 = getelementptr inbounds i32, i32* %74, i64 %76
store i32 %73, i32* %77
br label %78
78: ; preds = %67, %62
br label %79
79: ; preds = %78
%80 = load i32, i32* %14
%81 = add nsw i32 %80, 1
store i32 %81, i32* %14
br label %49
82: ; preds = %49
%83 = load i64*, i64** %17
%84 = call i32 @free(i64* %83)
store i64* null, i64** %17
%85 = load i32, i32* %16
store i32 %85, i32* %7
br label %86
86: ; preds = %82, %37
%87 = load i32, i32* %7
ret i32 %87
}

LLVM IR (-O2):
define i32 @kpt_relocate_BZ_grid_address([3 x i32]* %0, i32* nocapture writeonly %1, [3 x i32]* %2, i32* %3, [3 x double]* %4, i32* %5) {
%7 = load i32, i32* %3
%8 = getelementptr inbounds i32, i32* %3, i64 1
%9 = load i32, i32* %8
%10 = mul nsw i32 %9, %7
%11 = getelementptr inbounds i32, i32* %3, i64 2
%12 = load i32, i32* %11
%13 = mul nsw i32 %10, %12
%14 = shl i32 %13, 3
%15 = sext i32 %14 to i64
%16 = shl i32 %13, 6
%17 = tail call i64 @malloc(i32 %16)
%18 = inttoptr i64 %17 to i64*
%19 = icmp eq i64 %17, 0
br i1 %19, label %84, label %20
20: ; preds = %6
%21 = bitcast [3 x i32]* %0 to i32**
%22 = bitcast [3 x i32]* %2 to i32**
%23 = bitcast [3 x double]* %4 to double**
%24 = tail call i32 @kpt_relocate_dense_BZ_grid_address(i32** %21, i64* %18, i32** %22, i32* %3, double** %23, i32* %5)
%25 = icmp sgt i32 %13, 0
br i1 %25, label %26, label %82
26: ; preds = %20
%27 = call i32 @llvm.smax.i32(i32 %14, i32 1)
%28 = zext i32 %27 to i64
%29 = add nsw i64 %28, -1
%30 = and i64 %28, 1
%31 = icmp ult i64 %29, 3
br i1 %31, label %67, label %32
32: ; preds = %26
%33 = and i64 %28, 2147483640
br label %34
34: ; preds = %34, %32
%35 = phi i64 [ 0, %32 ], [ %64, %34 ]
%36 = phi i64 [ 0, %32 ], [ %65, %34 ]
%37 = getelementptr inbounds i64, i64* %18, i64 %35
%38 = load i64, i64* %37
%39 = icmp eq i64 %38, %15
%40 = trunc i64 %38 to i32
%41 = select i1 %39, i32 -1, i32 %40
%42 = getelementptr inbounds i32, i32* %1, i64 %35
store i32 %41, i32* %42
%43 = or i64 %35, 1
%44 = getelementptr inbounds i64, i64* %18, i64 %43
%45 = load i64, i64* %44
%46 = icmp eq i64 %45, %15
%47 = trunc i64 %45 to i32
%48 = select i1 %46, i32 -1, i32 %47
%49 = getelementptr inbounds i32, i32* %1, i64 %43
store i32 %48, i32* %49
%50 = or i64 %35, 2
%51 = getelementptr inbounds i64, i64* %18, i64 %50
%52 = load i64, i64* %51
%53 = icmp eq i64 %52, %15
%54 = trunc i64 %52 to i32
%55 = select i1 %53, i32 -1, i32 %54
%56 = getelementptr inbounds i32, i32* %1, i64 %50
store i32 %55, i32* %56
%57 = or i64 %35, 3
%58 = getelementptr inbounds i64, i64* %18, i64 %57
%59 = load i64, i64* %58
%60 = icmp eq i64 %59, %15
%61 = trunc i64 %59 to i32
%62 = select i1 %60, i32 -1, i32 %61
%63 = getelementptr inbounds i32, i32* %1, i64 %57
store i32 %62, i32* %63
%64 = add nuw nsw i64 %35, 4
%65 = add i64 %36, 4
%66 = icmp eq i64 %65, %33
br i1 %66, label %67, label %34
67: ; preds = %34, %26
%68 = phi i64 [ 0, %26 ], [ %64, %34 ]
%69 = icmp eq i64 %30, 0
br i1 %69, label %82, label %70
70: ; preds = %67, %70
%71 = phi i64 [ %79, %70 ], [ %68, %67 ]
%72 = phi i64 [ %80, %70 ], [ 0, %67 ]
%73 = getelementptr inbounds i64, i64* %18, i64 %71
%74 = load i64, i64* %73
%75 = icmp eq i64 %74, %15
%76 = trunc i64 %74 to i32
%77 = select i1 %75, i32 -1, i32 %76
%78 = getelementptr inbounds i32, i32* %1, i64 %71
store i32 %77, i32* %78
%79 = add nuw nsw i64 %71, 1
%80 = add i64 %72, 1
%81 = icmp eq i64 %80, %30
br i1 %81, label %82, label %70
82: ; preds = %67, %70, %20
%83 = tail call i32 @free(i64* %18)
br label %84
84: ; preds = %6, %82
%85 = phi i32 [ %24, %82 ], [ 0, %6 ]
ret i32 %85
}