Mercurial > hg > sv-dependency-builds
diff src/fftw-3.3.3/simd-support/avx.c @ 10:37bf6b4a2645
Add FFTW3
author | Chris Cannam |
---|---|
date | Wed, 20 Mar 2013 15:35:50 +0000 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/fftw-3.3.3/simd-support/avx.c Wed Mar 20 15:35:50 2013 +0000 @@ -0,0 +1,62 @@ +/* + * Copyright (c) 2003, 2007-11 Matteo Frigo + * Copyright (c) 2003, 2007-11 Massachusetts Institute of Technology + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + * + */ + + +#include "ifftw.h" + +#if HAVE_AVX + +#if defined(__x86_64__) || defined(_M_X64) || defined(_M_AMD64) + +#include "amd64-cpuid.h" + +int X(have_simd_avx)(void) +{ + static int init = 0, res; + + if (!init) { + res = 1 + && ((cpuid_ecx(1) & 0x18000000) == 0x18000000) + && ((xgetbv_eax(0) & 0x6) == 0x6); + init = 1; + } + return res; +} + +#else /* 32-bit code */ + +#include "x86-cpuid.h" + +int X(have_simd_avx)(void) +{ + static int init = 0, res; + + if (!init) { + res = !is_386() + && has_cpuid() + && ((cpuid_ecx(1) & 0x18000000) == 0x18000000) + && ((xgetbv_eax(0) & 0x6) == 0x6); + init = 1; + } + return res; +} +#endif + +#endif