提交 3f536fb7 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add a way to set properties on a context.

上级 caa214c7
......@@ -20,6 +20,7 @@ except ImportError:
pygpu = None
_context_reg = {}
_props_map = {}
def move_to_gpu(data):
......@@ -68,6 +69,7 @@ def reg_context(name, ctx):
if not isinstance(ctx, gpuarray.GpuContext):
raise TypeError("context is not GpuContext")
_context_reg[name] = ctx
_props_map[ctx] = dict()
def get_context(name):
......@@ -89,6 +91,19 @@ def get_context(name):
return _context_reg[name]
def _get_props(name):
ctx = get_context(name)
return _props_map[ctx]
def get_prop(name, k):
return _get_props(name)[k]
def set_prop(name, k, v):
_get_props(name)[k] = v
def list_contexts():
"""
Return an iterable of all the registered context names.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论