Skip to content

P
pytensor
提交 3eb0c5c5 authored 作者: João Victor Risso's avatar João Victor Risso
浏览文件
操作

Use npy_int64 type to obtain each dimension from grid

Signed-off-by: 's avatarJoão Victor Risso <joaovictor.risso@gmail.com>
上级 63bde71e
隐藏空白字符变更
内嵌 并排
正在显示 包含 4 行增加 和 4 行删除
theano/gpuarray/c_code/dnn_sptf.c
浏览文件 @ 3eb0c5c5
......@@ -102,10 +102,10 @@ APPLY_SPECIFIC(dnn_sptf)(PyGpuArrayObject * input,
}
// Obtain grid dimensions
num_images = (int) *( (npy_int *) PyArray_GETPTR1( grid_dims, 0 ) );
num_channels = (int) *( (npy_int *) PyArray_GETPTR1( grid_dims, 1 ) );
height = (int) *( (npy_int *) PyArray_GETPTR1( grid_dims, 2 ) );
width = (int) *( (npy_int *) PyArray_GETPTR1( grid_dims, 3 ) );
num_images = (int) *( (npy_int64 *) PyArray_GETPTR1( grid_dims, 0 ) );
num_channels = (int) *( (npy_int64 *) PyArray_GETPTR1( grid_dims, 1 ) );
height = (int) *( (npy_int64 *) PyArray_GETPTR1( grid_dims, 2 ) );
width = (int) *( (npy_int64 *) PyArray_GETPTR1( grid_dims, 3 ) );
gpu_grid_dims[0] = num_images;
gpu_grid_dims[1] = height;
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论