提交 afafe9b6 authored 作者: Frederic's avatar Frederic

Add missing include in the new gpuaelemwise op.

It was causing compilation crash in the DLT mlp
上级 83b2068b
......@@ -149,6 +149,8 @@ class GpuElemwise(HideC, Elemwise):
#define ga_double double
#define ga_half uint16_t
#include <Python.h>
#include <numpy/npy_common.h>
"""
return ElemwiseKernel(None, inps+outs, kop, preamble=support_code)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论