Browse Source

Implement starpu_get_blas_elemsize

Cédric Augonnet 15 years ago
parent
commit
fda086c7ae
2 changed files with 6 additions and 0 deletions
  1. 1 0
      include/starpu-data-interfaces.h
  2. 5 0
      src/datawizard/interfaces/blas_interface.c

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

@@ -42,6 +42,7 @@ uint32_t starpu_get_blas_nx(starpu_data_handle handle);
 uint32_t starpu_get_blas_ny(starpu_data_handle handle);
 uint32_t starpu_get_blas_local_ld(starpu_data_handle handle);
 uintptr_t starpu_get_blas_local_ptr(starpu_data_handle handle);
+size_t starpu_get_blas_elemsize(starpu_data_handle handle);
 
 /* BLOCK interface for 3D dense blocks */
 typedef struct starpu_block_interface_s {

+ 5 - 0
src/datawizard/interfaces/blas_interface.c

@@ -219,6 +219,11 @@ uintptr_t starpu_get_blas_local_ptr(data_state *state)
 	return (state->interface[node].blas.ptr);
 }
 
+size_t starpu_get_blas_elemsize(data_state *state)
+{
+	return (state->interface[0].blas.elemsize);
+}
+
 /* memory allocation/deallocation primitives for the BLAS interface */
 
 /* returns the size of the allocated area */