#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(int *);
int get_grid_rank(void *, int, int *);
scalar_t__ malloc(int);
int get_grid_size(void *self, int grid_id, int *size) {
  int rank = -1;
  if (get_grid_rank(self, grid_id, &rank) == (1))
    return (1);
  {
    int *shape = (int *)malloc(sizeof(int) * rank);
    int i;
    *size = 1;
    for (i = 0; i < rank; i++)
      *size *= shape[i];
    free(shape);
  }
  return (0);
}


x86 assembly (clang, -O0):
<get_grid_size>:
push   %rbp
mov    %rsp,%rbp
sub    $0x40,%rsp
mov    %rdi,-0x10(%rbp)
mov    %esi,-0x14(%rbp)
mov    %rdx,-0x20(%rbp)
movl   $0xffffffff,-0x24(%rbp)
mov    -0x10(%rbp),%rdi
mov    -0x14(%rbp),%esi
lea    -0x24(%rbp),%rdx
call   1050 <get_grid_rank@plt>
cmp    $0x1,%eax
jne    116f <get_grid_size+0x3f>
movl   $0x1,-0x4(%rbp)
jmp    11d1 <get_grid_size+0xa1>
movslq -0x24(%rbp),%rax
shl    $0x2,%rax
mov    %eax,%edi
call   1040 <malloc@plt>
mov    %rax,-0x30(%rbp)
mov    -0x20(%rbp),%rax
movl   $0x1,(%rax)
movl   $0x0,-0x34(%rbp)
mov    -0x34(%rbp),%eax
cmp    -0x24(%rbp),%eax
jge    11c1 <get_grid_size+0x91>
mov    -0x30(%rbp),%rax
movslq -0x34(%rbp),%rcx
mov    (%rax,%rcx,4),%ecx
mov    -0x20(%rbp),%rax
imul   (%rax),%ecx
mov    %ecx,(%rax)
mov    -0x34(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x34(%rbp)
jmp    1193 <get_grid_size+0x63>
mov    -0x30(%rbp),%rdi
call   1030 <free@plt>
movl   $0x0,-0x4(%rbp)
mov    -0x4(%rbp),%eax
add    $0x40,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<get_grid_size>:
push   %rbx
sub    $0x10,%rsp
mov    %rdx,%rbx
movl   $0xffffffff,0xc(%rsp)
ff
lea    0xc(%rsp),%rdx
call   1050 <get_grid_rank@plt>
mov    %eax,%ecx
mov    $0x1,%eax
cmp    $0x1,%ecx
je     11e9 <get_grid_size+0xb9>
mov    0xc(%rsp),%edi
shl    $0x2,%edi
call   1040 <malloc@plt>
movl   $0x1,(%rbx)
mov    0xc(%rsp),%esi
test   %esi,%esi
jle    11df <get_grid_size+0xaf>
lea    -0x1(%rsi),%rdx
mov    %esi,%ecx
and    $0x3,%ecx
cmp    $0x3,%rdx
jae    118c <get_grid_size+0x5c>
mov    $0x1,%edx
xor    %edi,%edi
jmp    11c4 <get_grid_size+0x94>
and    $0xfffffffc,%esi
mov    $0x1,%edx
xor    %edi,%edi
cs nopw 0x0(%rax,%rax,1)
imul   (%rax,%rdi,4),%edx
mov    %edx,(%rbx)
imul   0x4(%rax,%rdi,4),%edx
mov    %edx,(%rbx)
imul   0x8(%rax,%rdi,4),%edx
mov    %edx,(%rbx)
imul   0xc(%rax,%rdi,4),%edx
mov    %edx,(%rbx)
add    $0x4,%rdi
cmp    %rdi,%rsi
jne    11a0 <get_grid_size+0x70>
test   %rcx,%rcx
je     11df <get_grid_size+0xaf>
lea    (%rax,%rdi,4),%rsi
xor    %edi,%edi
nop
imul   (%rsi,%rdi,4),%edx
mov    %edx,(%rbx)
add    $0x1,%rdi
cmp    %rdi,%rcx
jne    11d0 <get_grid_size+0xa0>
mov    %rax,%rdi
call   1030 <free@plt>
xor    %eax,%eax
add    $0x10,%rsp
pop    %rbx
ret

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

LLVM IR (-O2):
define i32 @get_grid_size(i8* %0, i32 %1, i32* nocapture %2) {
%4 = alloca i32
%5 = bitcast i32* %4 to i8*
store i32 -1, i32* %4
%6 = call i32 @get_grid_rank(i8* %0, i32 %1, i32* %4)
%7 = icmp eq i32 %6, 1
br i1 %7, label %60, label %8
8: ; preds = %3
%9 = load i32, i32* %4
%10 = shl i32 %9, 2
%11 = call i64 @malloc(i32 %10)
%12 = inttoptr i64 %11 to i32*
store i32 1, i32* %2
%13 = load i32, i32* %4
%14 = icmp sgt i32 %13, 0
br i1 %14, label %15, label %58
15: ; preds = %8
%16 = zext i32 %13 to i64
%17 = add nsw i64 %16, -1
%18 = and i64 %16, 3
%19 = icmp ult i64 %17, 3
br i1 %19, label %44, label %20
20: ; preds = %15
%21 = and i64 %16, 4294967292
br label %22
22: ; preds = %22, %20
%23 = phi i32 [ 1, %20 ], [ %40, %22 ]
%24 = phi i64 [ 0, %20 ], [ %41, %22 ]
%25 = phi i64 [ 0, %20 ], [ %42, %22 ]
%26 = getelementptr inbounds i32, i32* %12, i64 %24
%27 = load i32, i32* %26
%28 = mul nsw i32 %23, %27
store i32 %28, i32* %2
%29 = or i64 %24, 1
%30 = getelementptr inbounds i32, i32* %12, i64 %29
%31 = load i32, i32* %30
%32 = mul nsw i32 %28, %31
store i32 %32, i32* %2
%33 = or i64 %24, 2
%34 = getelementptr inbounds i32, i32* %12, i64 %33
%35 = load i32, i32* %34
%36 = mul nsw i32 %32, %35
store i32 %36, i32* %2
%37 = or i64 %24, 3
%38 = getelementptr inbounds i32, i32* %12, i64 %37
%39 = load i32, i32* %38
%40 = mul nsw i32 %36, %39
store i32 %40, i32* %2
%41 = add nuw nsw i64 %24, 4
%42 = add i64 %25, 4
%43 = icmp eq i64 %42, %21
br i1 %43, label %44, label %22
44: ; preds = %22, %15
%45 = phi i32 [ 1, %15 ], [ %40, %22 ]
%46 = phi i64 [ 0, %15 ], [ %41, %22 ]
%47 = icmp eq i64 %18, 0
br i1 %47, label %58, label %48
48: ; preds = %44, %48
%49 = phi i32 [ %54, %48 ], [ %45, %44 ]
%50 = phi i64 [ %55, %48 ], [ %46, %44 ]
%51 = phi i64 [ %56, %48 ], [ 0, %44 ]
%52 = getelementptr inbounds i32, i32* %12, i64 %50
%53 = load i32, i32* %52
%54 = mul nsw i32 %49, %53
store i32 %54, i32* %2
%55 = add nuw nsw i64 %50, 1
%56 = add i64 %51, 1
%57 = icmp eq i64 %56, %18
br i1 %57, label %58, label %48
58: ; preds = %44, %48, %8
%59 = call i32 @free(i32* %12)
br label %60
60: ; preds = %3, %58
%61 = phi i32 [ 0, %58 ], [ 1, %3 ]
ret i32 %61
}