Update mkl_distributions.cpp

Author: samir-nasibli

mkl_random/src/mkl_distributions.cpp

@@ -659,18 +659,11 @@ irk_noncentral_chisquare_vec(irk_state *state, npy_intp len, double *res, const
 
             mkl_free(pvec);
         } else {
-            double lambda;
-
             /* noncentral_chisquare(1, nonc) ~ (Z + sqrt(nonc))**2 for df == 1 */
-            err = vdRngGaussian(VSL_RNG_METHOD_GAUSSIAN_ICDF, state->stream, len, res, d_zero, d_one);
             loc = sqrt(nonc);
-
-            #pragma ivdep
-            for(i=0; i<len; i++) {
-                lambda = res[i] + loc;
-                /* squaring could result in an overflow */   
-                res[i] = lambda * lambda;
-	    }
+            err = vdRngGaussian(VSL_RNG_METHOD_GAUSSIAN_ICDF, state->stream, len, res, loc, d_one);
+            assert(err == VSL_STATUS_OK);
+            vmdSqrt(len, res, res, VML_HA);
         }
     }
 }