Method test_MeanSquareError_gpu

test/python/test_operation.py:584–585

(self)

    @unittest.skipIf(not singa_wrap.USE_CUDA, 'CUDA is not enabled')
    def test_MeanSquareError_gpu(self):
        self._MeanSquareError_helper(gpu_dev)

    def _Abs_helper(self, dev):
        X = np.array([0.8, -1.2, 3.3, -3.6, -0.5,