1 2 /* 3 * Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved. 4 * 5 * Licensed under the Apache License, Version 2.0 (the "License"); 6 * you may not use this file except in compliance with the License. 7 * You may obtain a copy of the License at 8 * 9 * http://www.apache.org/licenses/LICENSE-2.0 10 * 11 * Unless required by applicable law or agreed to in writing, software 12 * distributed under the License is distributed on an "AS IS" BASIS, 13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 * See the License for the specific language governing permissions and 15 * limitations under the License. 16 * 17 */ 18 19 20 #if !(defined _CPU) 21 #error: please define _CPU - specific suffix to a function name 22 #endif 23 24 #if !(defined _VL) 25 #error: please define _VL - Number of elements per vector register 26 #endif 27 28 29 #include <immintrin.h> 30 #define CONFIG 1 31 #if ((_VL) == (2)) 32 #include "helperavx2_128.h" 33 #elif ((_VL) == (4)) 34 #include "helperavx2.h" 35 #elif ((_VL) == (8)) 36 #include "helperavx512f.h" 37 #endif 38 39 40 #define _JOIN4(a,b,c,d) a##b##c##d 41 #define JOIN4(a,b,c,d) _JOIN4(a,b,c,d) 42 43 #define atan_d_vec JOIN4(__fd_atan_,_VL,_,_CPU) 44 45 extern "C" vdouble atan_d_vec(vdouble const); 46 47 #include <atan_d_vec.h> 48