瀏覽代碼

implement starpu_get_block_elemsize

Cédric Augonnet 15 年之前
父節點
當前提交
9e9015d718
共有 2 個文件被更改,包括 7 次插入0 次删除
  1. 1 0
      include/starpu-data-interfaces.h
  2. 6 0
      src/datawizard/interfaces/block_interface.c

+ 1 - 0
include/starpu-data-interfaces.h

@@ -64,6 +64,7 @@ uint32_t starpu_get_block_nz(starpu_data_handle handle);
 uint32_t starpu_get_block_local_ldy(starpu_data_handle handle);
 uint32_t starpu_get_block_local_ldz(starpu_data_handle handle);
 uintptr_t starpu_get_block_local_ptr(starpu_data_handle handle);
+size_t starpu_get_block_elemsize(starpu_data_handle handle);
 
 /* vector interface for contiguous (non-strided) buffers */
 typedef struct starpu_vector_interface_s {

+ 6 - 0
src/datawizard/interfaces/block_interface.c

@@ -228,6 +228,12 @@ uintptr_t starpu_get_block_local_ptr(data_state *state)
 	return (state->interface[node].block.ptr);
 }
 
+size_t starpu_get_block_elemsize(data_state *state)
+{
+	return (state->interface[0].block.elemsize);
+}
+
+
 /* memory allocation/deallocation primitives for the BLOCK interface */
 
 /* returns the size of the allocated area */