Skip to main content

XOR (Exclusive OR) for branchless coding

The following example shows the array reversing using the XOR operator. No need to take any additional variable to reverse the array. 
int main(int argc, _TCHAR* argv[])
{
    char str[] = "I AM STUDENT";
    int length = strlen(str);
    for(int i = 0; i < ((length/2)); i++)
    {
        str[i] ^= str[length - (1+i)];
        str[length - (1+i)] ^= str[i];
        str[i] ^= str[length - (1+i)];
    }
    cout << str << endl;
    return 0;
}

The above example is one of the uses of XOR but XOR comes in handy when we can do branchless coding methods like butterfly switch etc. Sometimes this is very effective in speeding up the execution. Let's see one of the uses of XOR in branchless coding. I am taking a simple example of Y = | X |. Yes, I am generating abs of a supplied number. So, my function signature/definition in C++ looks like below:

int absoluteBranch(int x)
{
    if (x < 0)
{
        return -x;
    }
    else
{
        return x;
    }
}
From the C++ code, we can see the branching in code and until runtime, we can't 
definitely say which branch could be executed. If we look at the assembly generated 
by the code also shows branching (Without optimization):
absoluteBranch(int):
        push    rbp
        mov     rbp, rsp
        mov     DWORD PTR [rbp-4], edi
        cmp     DWORD PTR [rbp-4], 0
        jns     .L4
        mov     eax, DWORD PTR [rbp-4]
        neg     eax
        jmp     .L5
.L4:
        mov     eax, DWORD PTR [rbp-4]
.L5:
        pop     rbp

        ret

We have instructions like cmp(compare), jns(jump if not signed) and
jmp (unconditional jump). With the help of XOR we can completely remove
this branching of code while calculating abosolute of a signed number.

int absoluteBranchless(int x)
{
    int y = x >> (sizeof(int) * 8 - 1);
    return (x ^ y) - y;
}

Now I don't have any branch in the C++ code and assembly generated out of
it also branchless. Here goes the assembly (Without optimization):

absoluteBranchless(int):
        push    rbp
        mov     rbp, rsp
        mov     DWORD PTR [rbp-20], edi
        mov     eax, DWORD PTR [rbp-20]
        sar     eax, 31
        mov     DWORD PTR [rbp-4], eax
        mov     eax, DWORD PTR [rbp-20]
        xor     eax, DWORD PTR [rbp-4]
        sub     eax, DWORD PTR [rbp-4]
        pop     rbp
        ret


The SAR instruction shifts the MSB and it became all FFs. For positive number it's all 00s.
So for negative number mask became -1 (in two's complement) and for positive number
it will always be 0. We then XOR the number with mask and subtract with mask, which
effectively adds +1 in case of negative number and +0 for positive number.
In above example variable y is the mask value.

It's nice, we have no branching!

Comments

Remember you have shown this in office and we had some discussions on this as well.

If you cannot afford any extra memory for a new array then this XOR-ing is a good solution.I believe its used in hardware which cannot afford huge memory.

Popular posts from this blog

strcpy / strcmp implementation

 In C/C++ we have a library function called strcpy to copy the source character array to the destination character array. The C++ function details have been documented here . Today I tried to implement it in my own way. I have put forth a few conditions to implement this function, which are described below. 1. I don't want to pass the size of arrays as function parameters. I mean, the parameter gets passed implicitly 2. I don't want the program shall compile if any of the array sizes or both array sizes are zero or one. 3. There shall be no operation if source and destination arrays are the same. 4. No overflow happens if the destination array size is smaller than the source. 5. The destination array must be null-terminated after a successful copy. Below is the client code to test the implementation: int main() { // Case 1: Source and destination arrays are of the same size char src[] = "Hello World"; char dest[12] = {}; // n - 1 chars will be copied and the las...

Reversing char array without splitting the array to tokens

 I was reading about strdup, a C++ function and suddenly an idea came to my mind if this can be leveraged to aid in reversing a character array without splitting the array into words and reconstructing it again by placing spaces and removing trailing spaces. Again, I wanted an array to be passed as a function argument and an array size to be passed implicitly with the array to the function. Assumed, a well-formed char array has been passed into the function. No malformed array checking is done inside the function. So, the function signature and definition are like below: Below is the call from the client code to reverse the array without splitting tokens and reconstructing it. Finally, copy the reversed array to the destination.  For GNU C++, we should use strdup instead _strdup . On run, we get the following output: Demo code