Add openmp to elemwise op