int sqrt_custom(int a_nInput) {
  int op = a_nInput;
  int res = 0;
  int one = 1uL << 30;
  while (one > op) {
    one >>= 2;
  }
  while (one != 0) {
    if (op >= res + one) {
      op = op - (res + one);
      res = res + 2 * one;
    }
    res >>= 1;
    one >>= 2;
  }
  return res;
}


x86 assembly (clang, -O0):
<sqrt_custom>:
push   %rbp
mov    %rsp,%rbp
mov    %edi,-0x4(%rbp)
mov    -0x4(%rbp),%eax
mov    %eax,-0x8(%rbp)
movl   $0x0,-0xc(%rbp)
movl   $0x40000000,-0x10(%rbp)
mov    -0x10(%rbp),%eax
cmp    -0x8(%rbp),%eax
jle    1135 <sqrt_custom+0x35>
mov    -0x10(%rbp),%eax
sar    $0x2,%eax
mov    %eax,-0x10(%rbp)
jmp    111b <sqrt_custom+0x1b>
jmp    113a <sqrt_custom+0x3a>
cmpl   $0x0,-0x10(%rbp)
je     1188 <sqrt_custom+0x88>
mov    -0x8(%rbp),%eax
mov    -0xc(%rbp),%ecx
add    -0x10(%rbp),%ecx
cmp    %ecx,%eax
jl     1171 <sqrt_custom+0x71>
mov    -0x8(%rbp),%eax
mov    -0xc(%rbp),%ecx
add    -0x10(%rbp),%ecx
sub    %ecx,%eax
mov    %eax,-0x8(%rbp)
mov    -0xc(%rbp),%eax
mov    -0x10(%rbp),%ecx
shl    $0x1,%ecx
add    %ecx,%eax
mov    %eax,-0xc(%rbp)
mov    -0xc(%rbp),%eax
sar    $0x1,%eax
mov    %eax,-0xc(%rbp)
mov    -0x10(%rbp),%eax
sar    $0x2,%eax
mov    %eax,-0x10(%rbp)
jmp    113a <sqrt_custom+0x3a>
mov    -0xc(%rbp),%eax
pop    %rbp
ret

x86 assembly (clang, -O2):
<sqrt_custom>:
mov    $0x40000000,%eax
cs nopw 0x0(%rax,%rax,1)
nop
mov    %eax,%ecx
shr    $0x2,%eax
cmp    %edi,%ecx
jg     1110 <sqrt_custom+0x10>
test   %ecx,%ecx
je     1155 <sqrt_custom+0x55>
xor    %r8d,%r8d
xor    %esi,%esi
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax)
lea    (%rcx,%rsi,1),%edx
cmp    %edx,%edi
cmovl  %r8d,%edx
lea    (%rcx,%rcx,1),%eax
cmovl  %r8d,%eax
sub    %edx,%edi
add    %esi,%eax
sar    %eax
mov    %ecx,%edx
shr    $0x2,%edx
mov    %eax,%esi
cmp    $0x3,%ecx
mov    %edx,%ecx
ja     1130 <sqrt_custom+0x30>
ret
xor    %eax,%eax
ret

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

LLVM IR (-O2):
define i32 @sqrt_custom(i32 %0) {
br label %2
2: ; preds = %2, %1
%3 = phi i32 [ 1073741824, %1 ], [ %5, %2 ]
%4 = icmp sgt i32 %3, %0
%5 = lshr i32 %3, 2
br i1 %4, label %2, label %6
6: ; preds = %2
%7 = icmp eq i32 %3, 0
br i1 %7, label %22, label %8
8: ; preds = %6, %8
%9 = phi i32 [ %20, %8 ], [ %3, %6 ]
%10 = phi i32 [ %19, %8 ], [ 0, %6 ]
%11 = phi i32 [ %16, %8 ], [ %0, %6 ]
%12 = add nsw i32 %9, %10
%13 = icmp slt i32 %11, %12
%14 = shl nuw nsw i32 %9, 1
%15 = select i1 %13, i32 0, i32 %12
%16 = sub nsw i32 %11, %15
%17 = select i1 %13, i32 0, i32 %14
%18 = add nsw i32 %17, %10
%19 = ashr i32 %18, 1
%20 = lshr i32 %9, 2
%21 = icmp ult i32 %9, 4
br i1 %21, label %22, label %8
22: ; preds = %8, %6
%23 = phi i32 [ 0, %6 ], [ %19, %8 ]
ret i32 %23
}