提交 a5dda475 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix problems with output reuse in all gpuarrray blas ops.

上级 7e3db169
...@@ -19,10 +19,10 @@ except ImportError, e: ...@@ -19,10 +19,10 @@ except ImportError, e:
class BlasOp(HideC): class BlasOp(HideC):
def c_headers(self): def c_headers(self):
return ['<blas_api.h>'] return ['<blas_api.h>', '<numpy_compat.h>', '<gpuarray_helper.h>']
def c_header_dirs(self): def c_header_dirs(self):
return [pygpu.get_include()] return [pygpu.get_include(), os.path.dirname(__file__)]
def c_init_code(self): def c_init_code(self):
return ['import_pygpu__blas();'] return ['import_pygpu__blas();']
...@@ -286,10 +286,6 @@ class GpuDot22(BlasOp, Dot22): ...@@ -286,10 +286,6 @@ class GpuDot22(BlasOp, Dot22):
def c_code_cache_version(self): def c_code_cache_version(self):
return (3,) return (3,)
def c_header_dirs(self):
ret = super(GpuDot22, self).c_header_dirs()
return ret + [os.path.dirname(__file__)]
def c_headers(self): def c_headers(self):
ret = super(GpuDot22, self).c_headers() ret = super(GpuDot22, self).c_headers()
return ret + ['<numpy_compat.h>', '"gpuarray_helper.h"'] return ret + ['<numpy_compat.h>', '"gpuarray_helper.h"']
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论