提交 6e921145 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix 3d kernel.

上级 56333682
......@@ -95,7 +95,7 @@ KERNEL void im3d2col_kernel(const ga_size n,
const ga_size stride_h, const ga_size stride_w, const ga_size stride_d,
const ga_size height_col, const ga_size width_col, const ga_size depth_col,
GLOBAL_MEM DTYPE_INPUT_0 * data_col,
const ga_size data_im_offset) {
const ga_size offset_col) {
data_im = (GLOBAL_MEM DTYPE_INPUT_0 *)(((char *)data_im) + offset_im);
data_col = (GLOBAL_MEM DTYPE_INPUT_0 *)(((char *)data_col) + offset_col);
// grid stride looping
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论