amd/blis
1
0
Fork 0
mirror of https://github.com/amd/blis.git synced 2026-04-20 07:38:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Level-3 triangular routines now use different block sizes and kernels.

Browse Source 
Details:
    - Eliminated the need for override function in SUP for GEMMT/SYRK.
    - New set of block sizes, kernels and kernel preferences
      are added to cntx data structure for level-3 triangular routines.
    - Added supporting functions to set and get the above parameters from cntx.
    - Modified GEMMT/SYRK SUP code to use these new block sizes/kernels.
      In case they are not set, use the default block sizes/kernels of
      Level-3 SUP.

AMD-Internal: [CPUPL-3649]
Change-Id: Iee11bd4c4f1d8fbbb749c296258d1b8121c009a0
This commit is contained in:
Meghana Vankadari
2023-07-18 12:44:44 +05:30
parent 7b78d93282
commit 79e174ff0a
10 changed files with 571 additions and 159 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
ref_kernels/bli_cntx_ref.c
View File

@@ -5,7 +5,7 @@
libraries.
Copyright (C) 2014, The University of Texas at Austin
Copyright (C) 2018 - 2021, Advanced Micro Devices, Inc. All rights reserved.
Copyright (C) 2018 - 2023, Advanced Micro Devices, Inc. All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are
@@ -529,6 +529,79 @@ void GENBARNAME(cntx_init)
bli_mbool_init( &mbools[ BLIS_XXX ], TRUE, TRUE, TRUE, TRUE );
// -- Set level-3 small/unpacked micro-kernels, preferences and blocksizes
// for matrices dealing with triangular matrices-------------
// -- Set blocksizes -------------------------------------------------------
// s d c z
bli_blksz_init_easy( &blkszs[ BLIS_MR ], 0, 0, 0, 0 );
bli_blksz_init_easy( &blkszs[ BLIS_NR ], 0, 0, 0, 0 );
bli_blksz_init_easy( &blkszs[ BLIS_MC ], 0, 0, 0, 0 );
bli_blksz_init_easy( &blkszs[ BLIS_KC ], 0, 0, 0, 0 );
bli_blksz_init_easy( &blkszs[ BLIS_NC ], 0, 0, 0, 0 );
// Initialize the context with the default blocksize objects and their
// multiples.
bli_cntx_set_l3_sup_tri_blkszs
(
5,
// level-3
BLIS_KC, &blkszs[ BLIS_KC ],
BLIS_MC, &blkszs[ BLIS_MC ],
BLIS_NR, &blkszs[ BLIS_NR ],
BLIS_NC, &blkszs[ BLIS_NC ],
BLIS_MR, &blkszs[ BLIS_MR ],
cntx
);
funcs = bli_cntx_l3_sup_tri_kers_buf( cntx );
mbools = bli_cntx_l3_sup_tri_kers_prefs_buf( cntx );
#if 0
// Adhere to the small/unpacked ukernel mappings:
// - rv -> rrr, rcr
// - rg -> rrc, rcc
// - cv -> ccr, ccc
// - cg -> crr, crc
gen_sup_func_init( &funcs[ BLIS_RRR ],
&funcs[ BLIS_RCR ], gemmsup_rv_ukr_name );
gen_sup_func_init( &funcs[ BLIS_RRC ],
&funcs[ BLIS_RCC ], gemmsup_rg_ukr_name );
gen_sup_func_init( &funcs[ BLIS_CCR ],
&funcs[ BLIS_CCC ], gemmsup_cv_ukr_name );
gen_sup_func_init( &funcs[ BLIS_CRR ],
&funcs[ BLIS_CRC ], gemmsup_cg_ukr_name );
#endif
gen_func_init( &funcs[ BLIS_RRR ], gemmsup_rv_ukr_name );
gen_func_init( &funcs[ BLIS_RRC ], gemmsup_rv_ukr_name );
gen_func_init( &funcs[ BLIS_RCR ], gemmsup_rv_ukr_name );
gen_func_init( &funcs[ BLIS_RCC ], gemmsup_rv_ukr_name );
gen_func_init( &funcs[ BLIS_CRR ], gemmsup_rv_ukr_name );
gen_func_init( &funcs[ BLIS_CRC ], gemmsup_rv_ukr_name );
gen_func_init( &funcs[ BLIS_CCR ], gemmsup_rv_ukr_name );
gen_func_init( &funcs[ BLIS_CCC ], gemmsup_rv_ukr_name );
// Register the general-stride/generic ukernel to the "catch-all" slot
// associated with the BLIS_XXX enum value. This slot will be queried if
// *any* operand is stored with general stride.
gen_func_init( &funcs[ BLIS_XXX ], gemmsup_gx_ukr_name );
// Set the l3 sup ukernel storage preferences.
// s d c z
bli_mbool_init( &mbools[ BLIS_RRR ], TRUE, TRUE, TRUE, TRUE );
bli_mbool_init( &mbools[ BLIS_RRC ], TRUE, TRUE, TRUE, TRUE );
bli_mbool_init( &mbools[ BLIS_RCR ], TRUE, TRUE, TRUE, TRUE );
bli_mbool_init( &mbools[ BLIS_RCC ], TRUE, TRUE, TRUE, TRUE );
bli_mbool_init( &mbools[ BLIS_CRR ], TRUE, TRUE, TRUE, TRUE );
bli_mbool_init( &mbools[ BLIS_CRC ], TRUE, TRUE, TRUE, TRUE );
bli_mbool_init( &mbools[ BLIS_CCR ], TRUE, TRUE, TRUE, TRUE );
bli_mbool_init( &mbools[ BLIS_CCC ], TRUE, TRUE, TRUE, TRUE );
bli_mbool_init( &mbools[ BLIS_XXX ], TRUE, TRUE, TRUE, TRUE );
// -- Set level-1f kernels -------------------------------------------------
funcs = bli_cntx_l1f_kers_buf( cntx );