python类get_max_workspace_size()的实例源码

test_cuda.py 文件源码 项目：chainer-deconv 作者: germanRos 项目源码 文件源码 阅读 26 收藏 0 点赞 0 评论 0

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

test_cuda.py 文件源码 项目：chainer-deconv 作者: germanRos 项目源码 文件源码 阅读 41 收藏 0 点赞 0 评论 0

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