Skip to content

P
pytensor
提交 b47b95c3 authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier
浏览文件
操作

Convert CudaNdarray_DEV_DATA to PyGpuArray_DATA in ops GpuSoftmax and GpuSoftmaxWithBias.

上级 1c0aff18
隐藏空白字符变更
内嵌 并排
正在显示 包含 10 行增加 和 10 行删除
theano/sandbox/gpuarray/nnet.py
浏览文件 @ b47b95c3
......@@ -521,11 +521,11 @@ class GpuSoftmax (Op):
PyGpuArray_DIMS(%(x)s)[0],
PyGpuArray_DIMS(%(x)s)[1],
CudaNdarray_DEV_DATA(%(x)s),
PyArray_DATA(%(x)s),
PyGpuArray_STRIDES(%(x)s)[0],
PyGpuArray_STRIDES(%(x)s)[1],
CudaNdarray_DEV_DATA(%(z)s),
PyArray_DATA(%(z)s),
PyGpuArray_STRIDES(%(z)s)[0],
PyGpuArray_STRIDES(%(z)s)[1]
);
......@@ -539,11 +539,11 @@ class GpuSoftmax (Op):
PyGpuArray_DIMS(%(x)s)[0],
PyGpuArray_DIMS(%(x)s)[1],
CudaNdarray_DEV_DATA(%(x)s),
PyArray_DATA(%(x)s),
PyGpuArray_STRIDES(%(x)s)[0],
PyGpuArray_STRIDES(%(x)s)[1],
CudaNdarray_DEV_DATA(%(z)s),
PyArray_DATA(%(z)s),
PyGpuArray_STRIDES(%(z)s)[0],
PyGpuArray_STRIDES(%(z)s)[1]
);
......@@ -703,14 +703,14 @@ class GpuSoftmaxWithBias (GpuOp):
PyGpuArray_DIMS(%(x)s)[0],
PyGpuArray_DIMS(%(x)s)[1],
CudaNdarray_DEV_DATA(%(x)s),
PyArray_DATA(%(x)s),
PyGpuArray_STRIDES(%(x)s)[0],
PyGpuArray_STRIDES(%(x)s)[1],
CudaNdarray_DEV_DATA(%(b)s),
PyArray_DATA(%(b)s),
PyGpuArray_STRIDES(%(b)s)[0],
CudaNdarray_DEV_DATA(%(z)s),
PyArray_DATA(%(z)s),
PyGpuArray_STRIDES(%(z)s)[0],
PyGpuArray_STRIDES(%(z)s)[1]
);
......@@ -724,14 +724,14 @@ class GpuSoftmaxWithBias (GpuOp):
PyGpuArray_DIMS(%(x)s)[0],
PyGpuArray_DIMS(%(x)s)[1],
CudaNdarray_DEV_DATA(%(x)s),
PyArray_DATA(%(x)s),
PyGpuArray_STRIDES(%(x)s)[0],
PyGpuArray_STRIDES(%(x)s)[1],
CudaNdarray_DEV_DATA(%(b)s),
PyArray_DATA(%(b)s),
PyGpuArray_STRIDES(%(b)s)[0],
CudaNdarray_DEV_DATA(%(z)s),
PyArray_DATA(%(z)s),
PyGpuArray_STRIDES(%(z)s)[0],
PyGpuArray_STRIDES(%(z)s)[1]
);
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论