Sfoglia il codice sorgente

Implement starpu_get_blas_elemsize

Cédric Augonnet 15 anni fa
parent
commit
fda086c7ae

+ 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 */