U mdm@sDdZddlZddlZddlZddlZddlmZddZddZdS) z,Helpers for OpenMP support during the build.N)compile_test_programcCsRt|dr|jd}n|jj}tjdkr.dgStjdkrLdtddkrLgSd gS) Ncompilerrwin32z/openmpdarwinZopenmpCPPFLAGSz-fopenmp)hasattrr __class____name__sysplatformosgetenv)rr\/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/sklearn/_build_utils/openmp_helpers.pyget_openmp_flags    rc CsdtjkrdStd}tdd}|dk rH|d}dd|D}t}d}z^t|||d }|rd |d krt |d d d }t ||k}ndtjkrd}nd}Wn*t k r}z d}|}W5d}~XYnX|s tdrt d|ntd}t ||S)z6Check whether OpenMP test code can be compiled and runZPYODIDE_PACKAGE_ABIFz #include #include int main(void) { #pragma omp parallel printf("nthreads=%d\n", omp_get_num_threads()); return 0; } LDFLAGSN cSsg|]}|dr|qS))z-Lz -Wl,-rpathz-lz-Wl,--sysroot=/) startswith).0flagrrr ?s z(check_openmp_support..) extra_preargsextra_postargsz nthreads=r=rZPYTHON_CROSSENVTZSKLEARN_FAIL_NO_OPENMPz0Failed to build scikit-learn with OpenMP supportas *********** * WARNING * *********** It seems that scikit-learn cannot be built with OpenMP. - Make sure you have followed the installation instructions: https://scikit-learn.org/dev/developers/advanced_installation.html - If your compiler supports OpenMP but you still see this message, please submit a bug report at: https://github.com/scikit-learn/scikit-learn/issues - The build will continue with OpenMP-based parallelism disabled. Note however that some estimators will run in sequential mode instead of leveraging thread-based parallelism. *** )renvirontextwrapdedentrstripsplitrrintlen Exceptionwarningswarn) coderrZopenmp_exceptionoutputZnthreadsZopenmp_supported exceptionmessagerrrcheck_openmp_support(sP     r*) __doc__rr rr$Zpre_build_helpersrrr*rrrrs