ce2884f0a7dc26087a5225bc92e39643920e3e16,GPy/kern/linear.py,linear,dpsi2_dmuS,#linear#Any#Any#Any#Any#Any#Any#,169

Before Change

        self._psi_computations(Z, mu, S)
        AZZA = self.ZA.T[:, None, :, None] * self.ZA[None, :, None, :]
        AZZA = AZZA + AZZA.swapaxes(1, 2)
        target_S += (dL_dpsi2[:, :, :, None] * self.ZA[None, :, None, :] * self.ZA[None, None, :, :]).sum(1).sum(1)
        dpsi2_dmu = (dL_dpsi2[:, :, :, None] * np.tensordot(mu, AZZA, (-1, 0))).sum(1).sum(1)
        target_mu += dpsi2_dmu

    def dpsi2_dZ(self, dL_dpsi2, Z, mu, S, target):

After Change

        //include <omp.h>
        //include <math.h>
        
        weave_options = {"headers"           : ["<omp.h>"],
                         "extra_compile_args": ["-fopenmp -O3"],  //-march=native"],
                         "extra_link_args"   : ["-lgomp"]}

        N,M,Q = mu.shape[0],Z.shape[0],mu.shape[1]
        weave.inline(code, support_code=support_code, libraries=["gomp"],
                     arg_names=["N","M","Q","mu","AZZA","AZZA_2","target_mu","target_S","dL_dpsi2"],

Instances