提交 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): ...@@ -149,6 +149,8 @@ class GpuElemwise(HideC, Elemwise):
#define ga_double double #define ga_double double
#define ga_half uint16_t #define ga_half uint16_t
#include <Python.h>
#include <numpy/npy_common.h>
""" """
return ElemwiseKernel(None, inps+outs, kop, preamble=support_code) return ElemwiseKernel(None, inps+outs, kop, preamble=support_code)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论