提交 61bf8215 authored 作者: Frederic Bastien's avatar Frederic Bastien

cache the gpu code for some op.

上级 70c7de39
......@@ -435,7 +435,7 @@ class GpuElemwise(Op):
return sio.getvalue()
def c_code_cache_version(self):
return ()
return (1,0)
class GpuDimShuffle(Op):
def __init__(self, input_broadcastable, new_order):
......@@ -613,6 +613,9 @@ class GpuDimShuffle(Op):
sys.exit()
return sio.getvalue()
def c_code_cache_version(self):
return (1,0)
class GpuSum(Op):
def __init__(self, reduce_mask):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论