@@ -92,6 +92,28 @@ def __init__(self, *args, **kwargs):
9292 self .cfg .template_values ['cudaarch' ] = cudaarch
9393 self .cfg .generate_template_values ()
9494
95+ # Specify CUDA custom values for module load environment
96+ # The dirs should be in the order ['open64/bin', 'bin']
97+ bin_path = []
98+ if LooseVersion (self .version ) < LooseVersion ('7' ):
99+ bin_path .append (os .path .join ('open64' , 'bin' ))
100+ bin_path .append ('bin' )
101+
102+ lib_path = ['lib64' ]
103+ inc_path = ['include' ]
104+ if LooseVersion (self .version ) >= LooseVersion ('7' ):
105+ lib_path .append (os .path .join ('extras' , 'CUPTI' , 'lib64' ))
106+ inc_path .append (os .path .join ('extras' , 'CUPTI' , 'include' ))
107+ bin_path .append (os .path .join ('nvvm' , 'bin' ))
108+ lib_path .append (os .path .join ('nvvm' , 'lib64' ))
109+ inc_path .append (os .path .join ('nvvm' , 'include' ))
110+
111+ self .module_load_environment .CPATH = inc_path
112+ self .module_load_environment .LD_LIBRARY_PATH = lib_path
113+ self .module_load_environment .LIBRARY_PATH = lib_path + [os .path .join ('stubs' , 'lib64' )]
114+ self .module_load_environment .PATH = bin_path
115+ self .module_load_environment .PKG_CONFIG_PATH = ['pkgconfig' ]
116+
95117 def fetch_step (self , * args , ** kwargs ):
96118 """Check for EULA acceptance prior to getting sources."""
97119 # EULA for CUDA must be accepted via --accept-eula-for EasyBuild configuration option,
@@ -334,33 +356,3 @@ def make_module_extra(self):
334356 txt += self .module_generator .set_environment ('CUDA_PATH' , self .installdir )
335357 self .log .debug ("make_module_extra added this: %s" , txt )
336358 return txt
337-
338- def make_module_req_guess (self ):
339- """Specify CUDA custom values for PATH etc."""
340-
341- guesses = super (EB_CUDA , self ).make_module_req_guess ()
342-
343- # The dirs should be in the order ['open64/bin', 'bin']
344- bin_path = []
345- if LooseVersion (self .version ) < LooseVersion ('7' ):
346- bin_path .append (os .path .join ('open64' , 'bin' ))
347- bin_path .append ('bin' )
348-
349- lib_path = ['lib64' ]
350- inc_path = ['include' ]
351- if LooseVersion (self .version ) >= LooseVersion ('7' ):
352- lib_path .append (os .path .join ('extras' , 'CUPTI' , 'lib64' ))
353- inc_path .append (os .path .join ('extras' , 'CUPTI' , 'include' ))
354- bin_path .append (os .path .join ('nvvm' , 'bin' ))
355- lib_path .append (os .path .join ('nvvm' , 'lib64' ))
356- inc_path .append (os .path .join ('nvvm' , 'include' ))
357-
358- guesses .update ({
359- 'CPATH' : inc_path ,
360- 'LD_LIBRARY_PATH' : lib_path ,
361- 'LIBRARY_PATH' : lib_path + [os .path .join ('stubs' , 'lib64' )],
362- 'PATH' : bin_path ,
363- 'PKG_CONFIG_PATH' : ['pkgconfig' ],
364- })
365-
366- return guesses
0 commit comments