瀏覽代碼

tensor: fix report of data amount

Samuel Thibault 5 年之前
父節點
當前提交
d22c21736b
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      src/datawizard/interfaces/tensor_interface.c

+ 2 - 2
src/datawizard/interfaces/tensor_interface.c

@@ -713,7 +713,7 @@ static int copy_opencl_common(void *src_interface, unsigned src_node, void *dst_
                 }
         }
 
-	starpu_interface_data_copy(src_node, dst_node, src_block->nx*src_block->ny*src_block->nz*src_block->elemsize);
+	starpu_interface_data_copy(src_node, dst_node, src_block->nx*src_block->ny*src_block->nz*src_block->nt*src_block->elemsize);
 
 	return ret;
 }
@@ -815,7 +815,7 @@ static int copy_any_to_any(void *src_interface, unsigned src_node, void *dst_int
 		}
 	}
 
-	starpu_interface_data_copy(src_node, dst_node, nx*ny*nz*elemsize);
+	starpu_interface_data_copy(src_node, dst_node, nx*ny*nz*nt*elemsize);
 
 	return ret;
 }