提交 e6d2f0c7 authored 作者: Frederic Bastien's avatar Frederic Bastien

added version of the gpu code

上级 8e72e6c0
......@@ -192,6 +192,9 @@ class GpuConv(Op):
def c_headers(self):
return ['cuda_ndarray.cuh','<stdio.h>']
def c_code_cache_version(self):
return (0,1)
def c_support_code_apply(self, node, nodename):
return open(os.path.join(os.path.split(__file__)[0],'conv_kernel.cu')).read()+\
open(os.path.join(os.path.split(__file__)[0],'conv_full_kernel.cu')).read()+\
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论