我们从Python开源项目中,提取了以下2个代码示例,用于说明如何使用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())