-
Notifications
You must be signed in to change notification settings - Fork 1
/
main_solve.py
30 lines (23 loc) · 785 Bytes
/
main_solve.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import os
import sys
sys.path.append('./Solve/')
from Solver import ParTaskParRunCluster
def TestParTaskParRun():
yaml_dir = 'YamlFiles'
json_dir = 'JsonFiles'
mat_dir = 'PetscMat'
batch_size = 3
num_task = 8
command = './Solve/rs -ksp_type {} -pc_type {} -mat_file {} -yaml_file {} \n'
mat_template = './MatData/scipy_csr{}.npz'
idx_list = list( range(1000))
tmp_list = list(range(1500,2500))
idx_list.extend(tmp_list)
a = ParTaskParRunCluster(json_dir,yaml_dir,mat_dir,batch_size,idx_list,num_task)
a.Process(mat_template)
script_file = 'solve.sh'
header = ['#!/bin/bash \n']
footer = ['echo finished !! \n']
a.GenerateScript(script_file,header,footer,command)
if __name__ == '__main__':
TestParTaskParRun()