Python chainer.cuda 模块，get_max_workspace_size() 实例源码

def setUp(self):
        self.space = cuda.get_max_workspace_size()

def test_size(self):
        size = 1024
        cuda.set_max_workspace_size(size)
        self.assertEqual(size, cuda.get_max_workspace_size())