Method test_linear_gpu

test/python/test_onnx.py:362–363

(self)

    @unittest.skipIf(not singa_api.USE_CUDA, 'CUDA is not enabled')
    def test_linear_gpu(self):
        self._linear_helper(gpu_dev)

    def _gemm_helper(self, dev):
        A = np.random.randn(2, 3).astype(np.float32)