U <ºcAã @sÞddlZddlZddlZddlZzddlZddlmZWnek rLYnXdZ dZ dZ dZ dZ dZd Ze d d ¡Ze ed d deddd e ed ddeddd e edd deddd iZdd„Zdd„ZedkrÚeƒdS)éN)ÚCommandsz&/tmp/pytorch_blas_compare_environmentsZ mkl_2020_3Z mkl_2020_0Z open_blasZeigen)z USE_CUDA=0z USE_ROCM=0)ZcffiZcmakeZ hypothesisZninjaZnumpyZpyyamlÚ setuptoolsZtyping_extensionsÚ SubEnvSpec)Úgeneric_installsÚspecial_installsÚenvironment_variablesÚexpected_blas_symbolsÚexpected_mkl_version©)Úintel)z mkl=2020.3zmkl-include=2020.3)zBLAS=MKL)Zmkl_blas_sgemmz2020.0.3)r )z mkl=2020.0zmkl-include=2020.0z2020.0.0)Zopenblas)z BLAS=OpenBLAS)Z exec_blascGs:tjjj|Ž\}}}|r6tdt|ƒ›d|›d|›ƒ‚|S)zConvenience method.z conda error: z retcode: Ú )ÚcondaÚcliZ python_apiÚ run_commandÚOSErrorÚstr)ÚargsÚstdoutÚstderrÚretcoder r úU/tmp/pip-unpacked-wheel-gikjz4vx/torch/utils/benchmark/examples/blas_compare_setup.pyÚ conda_runRsrcCstj t¡rtdƒt t¡t t¡tj ddtj  tj  t ¡¡d  d¡ ¡}t ¡D]¶\}}tj t|¡}td|›d|›dƒttjd d |d ƒtd ƒtjd |›dtjtjd}|jrètd|j  d¡›d|j  d¡›ƒ‚tdƒttjd |ft|jžŽ|jrB|j\}}td|›ƒttjd |d|f|žŽ|jrütdƒtjd |›dd |j¡›dtjtjd}|jr¨td|j  d¡›d|j  d¡›ƒ‚tjd |›ddtjtjdj  d¡ ¡  ¡}|jD]} | |ksÜt!| ›dƒ‚qÜtd|›dƒtjd |›d|›ddtjtjd} tdƒtjd |›ddtjtjd} | jr‚td| j  d¡›d| j  d¡›ƒ‚| j  d¡} t| ƒ|jD](} d | krœ| | ksœt!d!| ›ƒ‚qœ|j"D]} | | ksÌt!‚qÌ|j#dk rd"|j#›| kst!‚td#|›ƒqZdS)$Nz$Cleaning: removing old working root.zgit rev-parse --show-toplevelT)ÚshellÚcwdzutf-8zCreating env: z: (ú)z--no-default-packagesz--prefixzpython=3z"Testing that env can be activated:zsource activate )rrrz-Failed to source base environment: stdout: z stderr: zInstalling packages:z"Installing packages from channel: z-czSetting environment variables.z && conda env config vars set ú z/Failed to set environment variables: stdout: z && envz not in envszBuilding PyTorch for env: `ú`z && cd z# && python setup.py install --cmakezChecking configuration:a[ && python -c "import torch;from torch.utils.benchmark import Timer;print(torch.__config__.show());setup = 'x=torch.ones((128, 128));y=torch.ones((128, 128))';counts = Timer('torch.mm(x, y)', setup).collect_callgrind(collect_baseline=False);stats = counts.as_standardized().stats(inclusive=True);print(stats.filter(lambda l: 'blas' in l.lower()))"ZBLASz*PyTorch build did not respect `BLAS=...`: z'- Intel(R) Math Kernel Library Version zBuild complete: )$ÚosÚpathÚexistsÚ WORKING_ROOTÚprintÚshutilÚrmtreeÚmakedirsÚ subprocessÚ check_outputÚdirnameÚrealpathÚ__file__ÚdecodeÚstripÚSUB_ENVSÚitemsÚjoinrÚconda_commandsZCREATEÚrunÚPIPEÚ returncoderrrÚINSTALLÚ BASE_PKG_DEPSrrrÚ splitlinesÚAssertionErrorrr )Zgit_rootÚenv_nameZenv_specZenv_pathZ base_sourceZchannelZ channel_depsZenv_setZactual_env_varsÚeZ build_runZ check_runZcheck_run_stdoutÚsr r rÚmain[sÌ   ýüüüÿþý ýýûÿ üû  ú  óÿ     r:Ú__main__)Ú collectionsrr"r%Zconda.cli.python_apir rr/Ú ImportErrorr Z MKL_2020_3Z MKL_2020_0Z OPEN_BLASZEIGENZGENERIC_ENV_VARSr4Ú namedtuplerr,rr:Ú__name__r r r rÚsb ÿ ûûûï"