Method test_gemm_gpu

test/python/test_operation.py:2627–2628

(self)

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

    def globalaveragepool_channel_first(self, dev):
        X = np.array([[[