提交 1d2b3245 authored 作者: Vivek Kulkarni's avatar Vivek Kulkarni

Correcting alignment

上级 36c30634
...@@ -1776,7 +1776,6 @@ CudaNdarray_inplace_elemwise(PyObject* py_self, PyObject * py_other, operator_t ...@@ -1776,7 +1776,6 @@ CudaNdarray_inplace_elemwise(PyObject* py_self, PyObject * py_other, operator_t
return 0; return 0;
} }
/* /*
* We need this inplace Add to support IncSubTensor * We need this inplace Add to support IncSubTensor
* It returns py_self on success with an additional reference. Else NULL. * It returns py_self on success with an additional reference. Else NULL.
......
...@@ -490,7 +490,6 @@ DllExport PyObject * CudaNdarray_View(const CudaNdarray * self); ...@@ -490,7 +490,6 @@ DllExport PyObject * CudaNdarray_View(const CudaNdarray * self);
DllExport PyObject * CudaNdarray_inplace_add(PyObject* py_self, PyObject * py_other); DllExport PyObject * CudaNdarray_inplace_add(PyObject* py_self, PyObject * py_other);
DllExport PyObject * CudaNdarray_Subscript(PyObject * py_self, PyObject * key); DllExport PyObject * CudaNdarray_Subscript(PyObject * py_self, PyObject * key);
DllExport int CudaNdarray_inplace_elemwise(PyObject* py_self, PyObject * py_other, operator_t fct_nb); DllExport int CudaNdarray_inplace_elemwise(PyObject* py_self, PyObject * py_other, operator_t fct_nb);
DllExport void CudaNdarray_vector_add_fast(CudaNdarray* py_self, CudaNdarray *py_other, PyArrayObject *indices_arr);
// Ensures that *arr is a pointer to a contiguous ndarray of the specified // Ensures that *arr is a pointer to a contiguous ndarray of the specified
// dimensions. // dimensions.
......
...@@ -1353,6 +1353,7 @@ def local_gpualloc_memset_0(node): ...@@ -1353,6 +1353,7 @@ def local_gpualloc_memset_0(node):
new_out = GpuAlloc(memset_0=True)(*node.inputs) new_out = GpuAlloc(memset_0=True)(*node.inputs)
return [new_out] return [new_out]
def safe_to_gpu(x): def safe_to_gpu(x):
if (isinstance(x.type, tensor.TensorType) and if (isinstance(x.type, tensor.TensorType) and
x.type.dtype == 'float32'): x.type.dtype == 'float32'):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论