Просмотр исходного кода

implement starpu_get_block_elemsize

Cédric Augonnet лет назад: 15
Родитель
Сommit
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 */