Support OpenMP

Simplify the #if statements