floraachy/OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enable use of AVX512 microkernels with NVIDIA HPC from version 22.3

Browse Source
This commit is contained in:
Martin Kroeker
2023-07-26 19:39:11 +02:00
committed by GitHub
parent 849c8806b8
commit 2a62d2df96
29 changed files with 124 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
kernel/x86_64/sbgemv_n_microk_cooperlake.c
+4 -1
View File
@@ -26,7 +26,10 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*****************************************************************************/
/* need a new enough GCC for avx512 support */
#if (( defined(__GNUC__) && __GNUC__ >= 10 && defined(__AVX512BF16__)) || (defined(__clang__) && __clang_major__ >= 9))
#ifdef __NVCOMPILER
#define NVCOMPVERS ( __NVCOMPILER_MAJOR__ * 100 + __NVCOMPILER_MINOR__ )
#endif
#if (( defined(__GNUC__) && __GNUC__ >= 10 && defined(__AVX512BF16__)) || (defined(__clang__) && __clang_major__ >= 9)) || (defined(__NVCOMPILER) && NVCOMPVERS >= 2203 )
#define HAVE_SBGEMV_N_ACCL_KERNEL 1
#include "common.h"