@@ -275,6 +275,24 @@ def download_files_for_some_targets(targets):
275275 result = subprocess .run (command , cwd = DOWNLOAD_DIR , shell = True )
276276 assert result .returncode == 0
277277
278+ elif basename == 'ge-spmm-cuda' :
279+ if not os .path .isfile (f'{ srcDir } /data/snap/amazon0302/amazon0302.mtx' ):
280+ command = f'mkdir -p ./snap && cd ./snap && source ../downloadSNAP.sh'
281+ result = subprocess .run (command , cwd = f'{ srcDir } /data' , shell = True )
282+ assert result .returncode == 0
283+
284+ elif basename == 'rsmt-cuda' :
285+ if not os .path .isfile (f'{ srcDir } /newblue7.kraftwerk70.3d.80.20.82.m8.gr' ):
286+ command = f'wget --no-check-certificate http://www.ispd.cc/contests/08/benchmark/newblue7.kraftwerk70.3d.80.20.82.m8.gr.gz'
287+ result = subprocess .run (command , cwd = srcDir , shell = True )
288+ assert result .returncode == 0
289+
290+ elif basename == 'bmf-cuda' :
291+ if not os .path .isfile (f'{ srcDir } /data/MNIST.in' ):
292+ command = f'mkdir -p ./data && cd ./ data && wget --no-check-certificate https://github.com/Funatiq/cuBool/raw/refs/heads/master/data/MNIST.in'
293+ result = subprocess .run (command , cwd = f'{ srcDir } ' , shell = True )
294+ assert result .returncode == 0
295+
278296 # need to have python2 installed for this to work
279297 elif basename == 'heat2d-cuda' :
280298 if not os .path .isfile (f'{ srcDir } /data.txt' ):
@@ -482,6 +500,8 @@ def modify_exe_args_for_some_targets(targets:list):
482500 target ['exeArgs' ] = 'graph.csv 10000 output'
483501 elif (basename == 'atomicCost-cuda' ):
484502 target ['exeArgs' ] = '16 10'
503+ elif (basename == 'snicit-cuda' ):
504+ target ['exeArgs' ] = '-k C -p ./dataset'
485505
486506
487507 return targets
@@ -808,7 +828,7 @@ def execute_target(target:dict, kernelName:str):
808828 print ('executing command:' , exeCommand , flush = True )
809829
810830 # we print the stderr to the stdout for analysis
811- # 15 minute timeout for now?
831+ # 5 minute timeout for now?
812832 # cm-cuda goes over 10 mins to run!
813833 try :
814834 # lsqt gives us zombie processes that take up GPU and do not respect the timeout
0 commit comments