Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "6b221920d7542fc768244bd06ca919dffdd56ed0"
Unverified Commit fb6af16f authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

[BUG] Fix crash during Cython build (#500)

parent 06e22f1e
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
python/dgl/_ffi/_cython/base.pxi
View file @ fb6af16f
from ..base import DGLError from ..base import DGLError
from libcpp.vector cimport vector from libcpp.vector cimport vector
from libcpp cimport bool
from cpython.version cimport PY_MAJOR_VERSION from cpython.version cimport PY_MAJOR_VERSION
from cpython cimport pycapsule from cpython cimport pycapsule
from libc.stdint cimport int64_t, uint64_t, uint8_t, uint16_t from libc.stdint cimport int64_t, uint64_t, uint8_t, uint16_t
...@@ -56,6 +57,8 @@ cdef extern from "dgl/runtime/c_runtime_api.h": ...@@ -56,6 +57,8 @@ cdef extern from "dgl/runtime/c_runtime_api.h":
ctypedef int64_t dgl_index_t ctypedef int64_t dgl_index_t
ctypedef DLTensor* DLTensorHandle ctypedef DLTensor* DLTensorHandle
ctypedef DLTensor DGLArray
ctypedef DGLArray* CDGLArrayHandle
ctypedef void* DGLStreamHandle ctypedef void* DGLStreamHandle
ctypedef void* DGLRetValueHandle ctypedef void* DGLRetValueHandle
ctypedef void* DGLFunctionHandle ctypedef void* DGLFunctionHandle
...@@ -101,7 +104,7 @@ cdef extern from "dgl/runtime/c_runtime_api.h": ...@@ -101,7 +104,7 @@ cdef extern from "dgl/runtime/c_runtime_api.h":
int dtype_bits, int dtype_bits,
int dtype_lanes, int dtype_lanes,
bool is_create, bool is_create,
DGLArrayHandle* out) CDGLArrayHandle* out)
int DGLArrayFree(DLTensorHandle handle) int DGLArrayFree(DLTensorHandle handle)
int DGLArrayCopyFromTo(DLTensorHandle src, int DGLArrayCopyFromTo(DLTensorHandle src,
DLTensorHandle to, DLTensorHandle to,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment