提交 3044beb7 authored 作者: Frederic's avatar Frederic

Add comment.

上级 369af1ad
...@@ -262,6 +262,8 @@ class GpuGemm(GpuOp): ...@@ -262,6 +262,8 @@ class GpuGemm(GpuOp):
== CudaNdarray_HOST_DIMS(%(z_in)s)[1]) == CudaNdarray_HOST_DIMS(%(z_in)s)[1])
&& (CudaNdarray_HOST_STRIDES(%(z_out)s)[0] >= 0) && (CudaNdarray_HOST_STRIDES(%(z_out)s)[0] >= 0)
&& (CudaNdarray_HOST_STRIDES(%(z_out)s)[1] >= 0) && (CudaNdarray_HOST_STRIDES(%(z_out)s)[1] >= 0)
// The following condition is needed as this is a condition by cublas
// on the memory layout of the output it accepts.
&& ((CudaNdarray_HOST_DIMS(%(z_out)s)[0] <= 1) && ((CudaNdarray_HOST_DIMS(%(z_out)s)[0] <= 1)
|| (CudaNdarray_HOST_STRIDES(%(z_out)s)[0] == 1) || (CudaNdarray_HOST_STRIDES(%(z_out)s)[0] == 1)
|| (CudaNdarray_HOST_DIMS(%(z_out)s)[1] <= 1) || (CudaNdarray_HOST_DIMS(%(z_out)s)[1] <= 1)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论