提交 61ad2da0 authored 作者: abergeron's avatar abergeron

Merge pull request #1661 from nouiz/gpuaelemwise

Add missing include in the new gpuaelemwise op.
......@@ -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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论