#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 size_t uint64_t;

/* Variables and functions */
size_t get_offset(size_t ndim, size_t *shape, size_t *strides, uint64_t bits) {
  size_t i, off = 0;
  for (i = 0; i < ndim; ++i)
    off += (((bits) >> i) & 1) * strides[i] * shape[i];
  return off;
}


x86 assembly (clang, -O0):
<get_offset>:
push   %rbp
mov    %rsp,%rbp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %rdx,-0x18(%rbp)
mov    %rcx,-0x20(%rbp)
movq   $0x0,-0x30(%rbp)
movq   $0x0,-0x28(%rbp)
mov    -0x28(%rbp),%rax
cmp    -0x8(%rbp),%rax
jae    1174 <get_offset+0x74>
mov    -0x20(%rbp),%rax
mov    -0x28(%rbp),%rcx
shr    %cl,%rax
and    $0x1,%rax
mov    -0x18(%rbp),%rcx
mov    -0x28(%rbp),%rdx
imul   (%rcx,%rdx,8),%rax
mov    -0x10(%rbp),%rcx
mov    -0x28(%rbp),%rdx
imul   (%rcx,%rdx,8),%rax
add    -0x30(%rbp),%rax
mov    %rax,-0x30(%rbp)
mov    -0x28(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x28(%rbp)
jmp    1124 <get_offset+0x24>
mov    -0x30(%rbp),%rax
pop    %rbp
ret

x86 assembly (clang, -O2):
<get_offset>:
test   %rdi,%rdi
je     1115 <get_offset+0x15>
mov    %rcx,%r8
cmp    $0x1,%rdi
jne    1118 <get_offset+0x18>
xor    %eax,%eax
xor    %r9d,%r9d
jmp    116f <get_offset+0x6f>
xor    %eax,%eax
ret
mov    %rdi,%r10
and    $0xfffffffffffffffe,%r10
xor    %eax,%eax
xor    %r9d,%r9d
cs nopw 0x0(%rax,%rax,1)
xchg   %ax,%ax
mov    %r8,%r11
mov    %r9d,%ecx
shr    %cl,%r11
and    $0x1,%r11d
imul   (%rdx,%r9,8),%r11
imul   (%rsi,%r9,8),%r11
add    %rax,%r11
lea    0x1(%r9),%ecx
mov    %r8,%rax
shr    %cl,%rax
and    $0x1,%eax
imul   0x8(%rdx,%r9,8),%rax
imul   0x8(%rsi,%r9,8),%rax
add    %r11,%rax
add    $0x2,%r9
cmp    %r10,%r9
jne    1130 <get_offset+0x30>
test   $0x1,%dil
je     118c <get_offset+0x8c>
mov    %r9d,%ecx
shr    %cl,%r8
and    $0x1,%r8d
imul   (%rdx,%r9,8),%r8
imul   (%rsi,%r9,8),%r8
add    %r8,%rax
ret

LLVM IR (-O0):
define i64 @get_offset(i64 %0, i64* %1, i64* %2, i64 %3) {
%5 = alloca i64
%6 = alloca i64*
%7 = alloca i64*
%8 = alloca i64
%9 = alloca i64
%10 = alloca i64
store i64 %0, i64* %5
store i64* %1, i64** %6
store i64* %2, i64** %7
store i64 %3, i64* %8
store i64 0, i64* %10
store i64 0, i64* %9
br label %11
11: ; preds = %32, %4
%12 = load i64, i64* %9
%13 = load i64, i64* %5
%14 = icmp ult i64 %12, %13
br i1 %14, label %15, label %35
15: ; preds = %11
%16 = load i64, i64* %8
%17 = load i64, i64* %9
%18 = lshr i64 %16, %17
%19 = and i64 %18, 1
%20 = load i64*, i64** %7
%21 = load i64, i64* %9
%22 = getelementptr inbounds i64, i64* %20, i64 %21
%23 = load i64, i64* %22
%24 = mul i64 %19, %23
%25 = load i64*, i64** %6
%26 = load i64, i64* %9
%27 = getelementptr inbounds i64, i64* %25, i64 %26
%28 = load i64, i64* %27
%29 = mul i64 %24, %28
%30 = load i64, i64* %10
%31 = add i64 %30, %29
store i64 %31, i64* %10
br label %32
32: ; preds = %15
%33 = load i64, i64* %9
%34 = add i64 %33, 1
store i64 %34, i64* %9
br label %11
35: ; preds = %11
%36 = load i64, i64* %10
ret i64 %36
}

LLVM IR (-O2):
define i64 @get_offset(i64 %0, i64* nocapture %1, i64* nocapture %2, i64 %3) {
%5 = icmp eq i64 %0, 0
br i1 %5, label %52, label %6
6: ; preds = %4
%7 = and i64 %0, 1
%8 = icmp eq i64 %0, 1
br i1 %8, label %37, label %9
9: ; preds = %6
%10 = and i64 %0, -2
br label %11
11: ; preds = %11, %9
%12 = phi i64 [ 0, %9 ], [ %33, %11 ]
%13 = phi i64 [ 0, %9 ], [ %34, %11 ]
%14 = phi i64 [ 0, %9 ], [ %35, %11 ]
%15 = lshr i64 %3, %13
%16 = and i64 %15, 1
%17 = getelementptr inbounds i64, i64* %2, i64 %13
%18 = load i64, i64* %17
%19 = mul nuw nsw i64 %18, %16
%20 = getelementptr inbounds i64, i64* %1, i64 %13
%21 = load i64, i64* %20
%22 = mul i64 %19, %21
%23 = add i64 %22, %12
%24 = or i64 %13, 1
%25 = lshr i64 %3, %24
%26 = and i64 %25, 1
%27 = getelementptr inbounds i64, i64* %2, i64 %24
%28 = load i64, i64* %27
%29 = mul nuw nsw i64 %28, %26
%30 = getelementptr inbounds i64, i64* %1, i64 %24
%31 = load i64, i64* %30
%32 = mul i64 %29, %31
%33 = add i64 %32, %23
%34 = add nuw i64 %13, 2
%35 = add i64 %14, 2
%36 = icmp eq i64 %35, %10
br i1 %36, label %37, label %11
37: ; preds = %11, %6
%38 = phi i64 [ undef, %6 ], [ %33, %11 ]
%39 = phi i64 [ 0, %6 ], [ %33, %11 ]
%40 = phi i64 [ 0, %6 ], [ %34, %11 ]
%41 = icmp eq i64 %7, 0
br i1 %41, label %52, label %42
42: ; preds = %37
%43 = lshr i64 %3, %40
%44 = and i64 %43, 1
%45 = getelementptr inbounds i64, i64* %2, i64 %40
%46 = load i64, i64* %45
%47 = mul nuw nsw i64 %46, %44
%48 = getelementptr inbounds i64, i64* %1, i64 %40
%49 = load i64, i64* %48
%50 = mul i64 %47, %49
%51 = add i64 %50, %39
br label %52
52: ; preds = %42, %37, %4
%53 = phi i64 [ 0, %4 ], [ %38, %37 ], [ %51, %42 ]
ret i64 %53
}