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

Add a way to set properties on a context.

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