GMS_32f_atan_32f_avx512.h

#ifndef __GMS_32F_ATAN_32F_AVX512_H__
#define __GMS_32F_ATAN_32F_AVX512_H__ 260920211224

/*
    Based on VOLK project.
    Modified by Bernard Gingold on:
    Date: 26-09-2021 12:24PM +00200
    contact: beniekg@gmail.com
    Few modification were added to original
    implementation (ICC pragmas, alignment directives and code layout rescheduled,
    unrolling completely 2-iteration for-loops)
    Converted to AVX512 implementation
*/

/*
 * Copyright 2018 Free Software Foundation, Inc.
 *
 * This file is part of GNU Radio
 *
 * GNU Radio 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 3, or (at your option)
 * any later version.
 *
 * GNU Radio 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 GNU Radio; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */


#include <cstdint>
#include "GMS_config.h"


#if !defined(DSP_32F_ATAN_32F_AVX512_BLOCK)
    #define DSP_32F_ATAN_32F_AVX512_BLOCK                        \
       int32_t idx = 0;                                          \
       const int32_t len = npoints/16;                           \
       register __m512 aVal,pio2, x, y, z, arctangent;           \
       register __m512 n_third,p_third,fzeroes,fones;	         \
       register __m512 ftwos,ffours,t0;                          \
       __mmask16 condition = 0;                                  \
       pio2 = _mm512_set1_ps(1.5707963267948966192f);            \
       fzeroes = _mm512_setzero_ps();                            \
       t0 = fzeroes;                                             \
       fones = _mm512_set1_ps(1.0f);                             \
       ftwos = _mm512_set1_ps(2.0f);                             \
       ffours  = _mm512_set1_ps(4.0f);                           \
       n_third = _mm512_set1_ps(-0.3333333333333333333333333f);  \
       p_third = _mm512_set1_ps(0.3333333333333333333333333f);
#endif

     
         __ATTR_HOT__
	 __ATTR_ALIGN__(32)
         void dsp_32f_atan_32f_u_avx512_looped(float * __restrict b,
	                                       float * __restrict a,
				               const int32_t npoints); 


        
         __ATTR_HOT__
	 __ATTR_ALIGN__(32)
	  void dsp_32f_atan_32f_a_avx512_looped(float * __restrict __ATTR_ALIGN__(64) b,
	                                       float * __restrict __ATTR_ALIGN__(64) a,
				               const int32_t npoints);


	
	 __ATTR_HOT__
	 __ATTR_ALIGN__(32)
	 __ATTR_VECTORCALL__
	 __m512 dsp_32f_atan_32f_avx512(const __m512 v); 










#endif /*__GMS_32F_ATAN_32F_AVX512_H__*/