amd/blis
1
0
Fork 0
mirror of https://github.com/amd/blis.git synced 2026-04-19 23:28:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Enabled disable-sba-pools feature in AOCL-BLAS (#101)

Browse Source 
Co-authored-by: harsh dave <harsdave@amd.com>
This commit is contained in:
Dave, Harsh
2025-09-19 21:45:08 +05:30
committed by GitHub
parent ee3d250b7a
commit 7076294286
3 changed files with 17 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
CMakeLists.txt
View File

@@ -560,8 +560,7 @@ if(ENABLE_SBA_POOLS)
message(" Internal memory pools for small blocks are enabled.")
set(ENABLE_SBA_POOLS_01 1)
else()
#message(" Internal memory pools for small blocks are disabled.")
message(FATAL_ERROR "Disabling memory pools for small blocks is currently disabled, awaiting fixes to this functionality.")
message(" Internal memory pools for small blocks are disabled.")
set(ENABLE_SBA_POOLS_01 0)
endif()
cmake_print_variables(ENABLE_MEM_TRACING)

4
configure vendored
View File

@@ -3253,9 +3253,7 @@ main()
echo "${script_name}: internal memory pools for small blocks are enabled."
enable_sba_pools_01=1
else
#echo "${script_name}: internal memory pools for small blocks are disabled."
echo "${script_name}: *** disabling memory pools for small blocks is currently disabled, awaiting fixes to this functionality."
exit 1
echo "${script_name}: internal memory pools for small blocks are disabled."
enable_sba_pools_01=0
fi
if [ "x${enable_mem_tracing}" = "xyes" ]; then

26
frame/base/bli_sba.c
View File

@@ -47,12 +47,16 @@ apool_t* bli_sba_query( void )
void bli_sba_init( void )
{
#ifdef BLIS_ENABLE_SBA_POOLS
bli_apool_init( &sba );
#endif
}
void bli_sba_finalize( void )
{
#ifdef BLIS_ENABLE_SBA_POOLS
bli_apool_finalize( &sba );
#endif
}
void* bli_sba_acquire
@@ -97,7 +101,7 @@ void* bli_sba_acquire
}
#else
block = bli_malloc_intl( req_size, &r_val );
block = bli_malloc_user( req_size, &r_val );
#endif
@@ -141,7 +145,7 @@ void bli_sba_release
}
#else
bli_free_intl( block );
bli_free_user( block );
#endif
}
@@ -151,11 +155,11 @@ array_t* bli_sba_checkout_array
const siz_t n_threads
)
{
#ifndef BLIS_ENABLE_SBA_POOLS
#ifndef BLIS_ENABLE_SBA_POOLS
return NULL;
#endif
#else
return bli_apool_checkout_array( n_threads, &sba );
#endif
}
void bli_sba_checkin_array
@@ -163,11 +167,11 @@ void bli_sba_checkin_array
array_t* restrict array
)
{
#ifndef BLIS_ENABLE_SBA_POOLS
#ifndef BLIS_ENABLE_SBA_POOLS
return;
#endif
#else
bli_apool_checkin_array( array, &sba );
#endif
}
void bli_sba_rntm_set_pool
@@ -177,16 +181,16 @@ void bli_sba_rntm_set_pool
rntm_t* restrict rntm
)
{
#ifndef BLIS_ENABLE_SBA_POOLS
#ifndef BLIS_ENABLE_SBA_POOLS
bli_rntm_set_sba_pool( NULL, rntm );
return;
#endif
#else
// Query the pool_t* in the array_t corresponding to index.
pool_t* restrict pool = bli_apool_array_elem( index, array );
// Embed the pool_t* into the rntm_t.
bli_rntm_set_sba_pool( pool, rntm );
#endif
}