SSEdefs.h
#ifndef __SSEDEFS_H__
#define __SSEDEFS_H__
#include "cosmoType.h"
#if CMK_USE_AVX
#if !defined(__AVX__)
#undef CMK_USE_AVX
#define CMK_USE_AVX 0
#else
#warning "using AVX"
#endif
#endif
#if CMK_USE_SSE2 && !defined(__SSE2__)
#undef CMK_USE_SSE2
#define CMK_USE_SSE2 0
#endif
#if CMK_USE_AVX || CMK_USE_SSE2
#define CMK_SSE 1
#endif
#if CMK_USE_AVX
#ifdef COSMO_FLOAT
#error "single-precision AVX is not supported"
#else
#include "SSE-Double.h"
#define SSE_VECTOR_WIDTH 4
#define FORCE_INPUT_LIST_PAD 3
typedef SSEDouble SSEcosmoType;
#define SSELoad(where, arr, idx, field) where(arr[idx]field, arr[idx+1]field, arr[idx+2]field, arr[idx+3]field)
#define SSEStore(what, arr, idx, field) { \
double p[4]; \
storeu(p, what); \
arr[idx]field = p[0]; \
arr[idx+1]field = p[1]; \
arr[idx+2]field = p[2]; \
arr[idx+3]field = p[3]; \
}
enum {cosmoMask=0xf};
#endif
#elif CMK_USE_SSE2
#ifdef COSMO_FLOAT
#define SSE_COSMO_FLOAT
#if defined(__SSE2__)
#include "SSE-Float.h"
#define SSE_VECTOR_WIDTH 4
#define FORCE_INPUT_LIST_PAD 3
typedef SSEFloat SSEcosmoType;
#define SSELoad(where, arr, idx, field) where(arr[idx]field, arr[idx+1]field, arr[idx+2]field, arr[idx+3]field)
#define SSEStore(what, arr, idx, field) { \
float p[4]; \
storeu(p, what); \
arr[idx]field = p[0]; \
arr[idx+1]field = p[1]; \
arr[idx+2]field = p[2]; \
arr[idx+3]field = p[3]; \
}
enum {cosmoMask=0xf};
#else
#error("SSE not available");
#endif
#else
#if defined(__SSE2__) && !defined(SSE_COSMO_FLOAT)
#include "SSE-Double.h"
#define SSE_VECTOR_WIDTH 2
#define FORCE_INPUT_LIST_PAD 1
typedef SSEDouble SSEcosmoType;
#define SSELoad(where, arr, idx, field) where(arr[idx]field, arr[idx+1]field)
#define SSEStore(what, arr, idx, field) { \
storel(&arr[idx]field, what); \
storeh(&arr[idx+1]field, what); \
}
enum {cosmoMask=0x3};
#else
#error("SSE not available");
#endif
#endif
#endif
#endif