69 lines
1.5 KiB
C
69 lines
1.5 KiB
C
/*
|
|
* Copyright (C) 2022 NHR@FAU, University Erlangen-Nuremberg.
|
|
* All rights reserved. This file is part of MD-Bench.
|
|
* Use of this source code is governed by a LGPL-3.0
|
|
* license that can be found in the LICENSE file.
|
|
*/
|
|
#ifndef __SIMD_H__
|
|
#define __SIMD_H__
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <immintrin.h>
|
|
|
|
#ifndef NO_ZMM_INTRIN
|
|
# include <zmmintrin.h>
|
|
#endif
|
|
|
|
#ifndef CLUSTER_M
|
|
# define CLUSTER_M 1
|
|
#endif
|
|
|
|
#ifndef CLUSTER_N
|
|
# define CLUSTER_N 1
|
|
#endif
|
|
|
|
#if defined(__ISA_AVX512__)
|
|
# if PRECISION == 2
|
|
# include "simd/avx512_double.h"
|
|
# else
|
|
# include "simd/avx512_float.h"
|
|
# endif
|
|
#endif
|
|
|
|
#if defined(__ISA_AVX2__)
|
|
# if PRECISION == 2
|
|
# include "simd/avx2_double.h"
|
|
# else
|
|
# include "simd/avx2_float.h"
|
|
# endif
|
|
#endif
|
|
|
|
#if defined(__ISA_AVX__)
|
|
# if PRECISION == 2
|
|
# include "simd/avx_double.h"
|
|
# else
|
|
# include "simd/avx_float.h"
|
|
# endif
|
|
#endif
|
|
|
|
#define SIMD_PRINT_REAL(a) simd_print_real(#a, a);
|
|
#define SIMD_PRINT_MASK(a) simd_print_mask(#a, a);
|
|
|
|
static inline void simd_print_real(const char *ref, MD_SIMD_FLOAT a) {
|
|
double x[VECTOR_WIDTH];
|
|
memcpy(x, &a, sizeof(x));
|
|
|
|
fprintf(stdout, "%s: ", ref);
|
|
for(int i = 0; i < VECTOR_WIDTH; i++) {
|
|
fprintf(stdout, "%f ", x[i]);
|
|
}
|
|
|
|
fprintf(stdout, "\n");
|
|
}
|
|
|
|
static inline void simd_print_mask(const char *ref, MD_SIMD_MASK a) { fprintf(stdout, "%s: %x\n", ref, simd_mask_to_u32(a)); }
|
|
|
|
#endif // __SIMD_H__
|