Method test_matmul_gpu

test/python/test_onnx.py:266–267

(self)

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

    def _max_pool_helper(self, dev):
        x = tensor.Tensor(shape=(2, 3, 4, 4), device=dev)