提交 8ee15d0c authored 作者: notoraptor's avatar notoraptor

Prevent function redefinition in C code for dimshuffle.

上级 8dccbe6e
#section support_code_apply
int gpu_dimshuffle(PyGpuArrayObject* input, PyGpuArrayObject** out, PARAMS_TYPE* params) {
int APPLY_SPECIFIC(gpu_dimshuffle)(PyGpuArrayObject* input, PyGpuArrayObject** out, PARAMS_TYPE* params) {
PyGpuArrayObject *tmp = NULL;
npy_intp nd_in = PyArray_SIZE(params->input_broadcastable);
npy_intp nd_out = PyArray_SIZE(params->_new_order);
......
......@@ -409,7 +409,7 @@ class GpuDimShuffle(DimShuffle):
"""
_f16_ok = True
c_func_name = 'gpu_dimshuffle'
c_func_name = 'APPLY_SPECIFIC(gpu_dimshuffle)'
def make_node(self, input):
ctx_name = infer_context_name(input)
......
#section support_code_apply
int cpu_dimshuffle(PyArrayObject* input, PyArrayObject** res, PARAMS_TYPE* params) {
int APPLY_SPECIFIC(cpu_dimshuffle)(PyArrayObject* input, PyArrayObject** res, PARAMS_TYPE* params) {
npy_bool* input_broadcastable;
npy_int64* new_order;
npy_intp nd_in;
......
......@@ -131,7 +131,7 @@ class DimShuffle(COp):
check_input = False
__props__ = ("input_broadcastable", "new_order", "inplace")
c_func_file = 'c_code/dimshuffle.c'
c_func_name = 'cpu_dimshuffle'
c_func_name = 'APPLY_SPECIFIC(cpu_dimshuffle)'
@property
def params_type(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论