@@ -322,26 +322,17 @@ <h1>Source code for mache.spack.env</h1><div class="highlight"><pre>
322322 < span class ="k "> if</ span > < span class ="n "> config</ span > < span class ="o "> .</ span > < span class ="n "> has_option</ span > < span class ="p "> (</ span > < span class ="s1 "> 'spack'</ span > < span class ="p "> ,</ span > < span class ="s1 "> 'cray_compilers'</ span > < span class ="p "> ):</ span >
323323 < span class ="n "> cray_compilers</ span > < span class ="o "> =</ span > < span class ="n "> section</ span > < span class ="o "> .</ span > < span class ="n "> getboolean</ span > < span class ="p "> (</ span > < span class ="s1 "> 'cray_compilers'</ span > < span class ="p "> )</ span >
324324
325- < span class ="n "> mod_env_commands</ span > < span class ="o "> =</ span > < span class ="s1 "> 'module purge</ span > < span class ="se "> \n</ span > < span class ="s1 "> '</ span >
326-
327- < span class ="k "> for</ span > < span class ="n "> shell_filename</ span > < span class ="ow "> in</ span > < span class ="p "> [</ span >
328- < span class ="sa "> f</ span > < span class ="s1 "> '</ span > < span class ="si "> {</ span > < span class ="n "> machine</ span > < span class ="si "> }</ span > < span class ="s1 "> .</ span > < span class ="si "> {</ span > < span class ="n "> shell</ span > < span class ="si "> }</ span > < span class ="s1 "> '</ span > < span class ="p "> ,</ span >
329- < span class ="sa "> f</ span > < span class ="s1 "> '</ span > < span class ="si "> {</ span > < span class ="n "> machine</ span > < span class ="si "> }</ span > < span class ="s1 "> _</ span > < span class ="si "> {</ span > < span class ="n "> compiler</ span > < span class ="si "> }</ span > < span class ="s1 "> _</ span > < span class ="si "> {</ span > < span class ="n "> mpi</ span > < span class ="si "> }</ span > < span class ="s1 "> .</ span > < span class ="si "> {</ span > < span class ="n "> shell</ span > < span class ="si "> }</ span > < span class ="s1 "> '</ span > < span class ="p "> ,</ span >
330- < span class ="p "> ]:</ span >
331- < span class ="n "> path</ span > < span class ="o "> =</ span > < span class ="p "> (</ span >
332- < span class ="n "> importlib_resources</ span > < span class ="o "> .</ span > < span class ="n "> files</ span > < span class ="p "> (</ span > < span class ="s1 "> 'mache.spack.templates'</ span > < span class ="p "> )</ span > < span class ="o "> /</ span > < span class ="n "> shell_filename</ span >
333- < span class ="p "> )</ span >
334- < span class ="k "> try</ span > < span class ="p "> :</ span >
335- < span class ="k "> with</ span > < span class ="nb "> open</ span > < span class ="p "> (</ span > < span class ="nb "> str</ span > < span class ="p "> (</ span > < span class ="n "> path</ span > < span class ="p "> ))</ span > < span class ="k "> as</ span > < span class ="n "> fp</ span > < span class ="p "> :</ span >
336- < span class ="n "> template</ span > < span class ="o "> =</ span > < span class ="n "> Template</ span > < span class ="p "> (</ span > < span class ="n "> fp</ span > < span class ="o "> .</ span > < span class ="n "> read</ span > < span class ="p "> ())</ span >
337- < span class ="k "> except</ span > < span class ="ne "> FileNotFoundError</ span > < span class ="p "> :</ span >
338- < span class ="c1 "> # there's nothing to add, which is fine</ span >
339- < span class ="k "> continue</ span >
340- < span class ="n "> shell_script</ span > < span class ="o "> =</ span > < span class ="n "> template</ span > < span class ="o "> .</ span > < span class ="n "> render</ span > < span class ="p "> (</ span >
341- < span class ="n "> e3sm_lapack</ span > < span class ="o "> =</ span > < span class ="n "> include_e3sm_lapack</ span > < span class ="p "> ,</ span >
342- < span class ="n "> e3sm_hdf5_netcdf</ span > < span class ="o "> =</ span > < span class ="n "> include_e3sm_hdf5_netcdf</ span > < span class ="p "> ,</ span >
343- < span class ="p "> )</ span >
344- < span class ="n "> mod_env_commands</ span > < span class ="o "> =</ span > < span class ="sa "> f</ span > < span class ="s1 "> '</ span > < span class ="si "> {</ span > < span class ="n "> mod_env_commands</ span > < span class ="si "> }</ span > < span class ="se "> \n</ span > < span class ="si "> {</ span > < span class ="n "> shell_script</ span > < span class ="si "> }</ span > < span class ="s1 "> '</ span >
325+ < span class ="n "> mod_env_commands</ span > < span class ="o "> =</ span > < span class ="n "> get_spack_script</ span > < span class ="p "> (</ span >
326+ < span class ="n "> spack_path</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span >
327+ < span class ="n "> env_name</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span >
328+ < span class ="n "> compiler</ span > < span class ="o "> =</ span > < span class ="n "> compiler</ span > < span class ="p "> ,</ span >
329+ < span class ="n "> mpi</ span > < span class ="o "> =</ span > < span class ="n "> mpi</ span > < span class ="p "> ,</ span >
330+ < span class ="n "> shell</ span > < span class ="o "> =</ span > < span class ="n "> shell</ span > < span class ="p "> ,</ span >
331+ < span class ="n "> machine</ span > < span class ="o "> =</ span > < span class ="n "> machine</ span > < span class ="p "> ,</ span >
332+ < span class ="n "> load_spack_env</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> ,</ span >
333+ < span class ="n "> include_e3sm_lapack</ span > < span class ="o "> =</ span > < span class ="n "> include_e3sm_lapack</ span > < span class ="p "> ,</ span >
334+ < span class ="n "> include_e3sm_hdf5_netcdf</ span > < span class ="o "> =</ span > < span class ="n "> include_e3sm_hdf5_netcdf</ span > < span class ="p "> ,</ span >
335+ < span class ="p "> )</ span >
345336
346337 < span class ="n "> mpicc</ span > < span class ="p "> ,</ span > < span class ="n "> mpicxx</ span > < span class ="p "> ,</ span > < span class ="n "> mpifc</ span > < span class ="o "> =</ span > < span class ="n "> _get_mpi_compilers</ span > < span class ="p "> (</ span >
347338 < span class ="n "> machine</ span > < span class ="p "> ,</ span > < span class ="n "> compiler</ span > < span class ="p "> ,</ span > < span class ="n "> mpi</ span > < span class ="p "> ,</ span > < span class ="n "> cray_compilers</ span >
0 commit comments