xref: /petsc/src/tao/unconstrained/tutorials/rosenbrock4cu.cu (revision d756bedd70a89ca052be956bccd75c5761cb2ab4)
1ec9796c4SHansol Suh const char help[] = "CUDA backend of rosenbrock4cu.cu\n";
2ec9796c4SHansol Suh 
3ec9796c4SHansol Suh /* ------------------------------------------------------------------------
4ec9796c4SHansol Suh 
5ec9796c4SHansol Suh   Copy of rosenbrock1.c.
6f0b74427SPierre Jolivet   Once PETSc test harness supports conditional linking, we can remove this duplicate.
7ec9796c4SHansol Suh   See https://gitlab.com/petsc/petsc/-/issues/1173
8ec9796c4SHansol Suh   ------------------------------------------------------------------------- */
9ec9796c4SHansol Suh 
10ec9796c4SHansol Suh #include "rosenbrock4.h"
11ec9796c4SHansol Suh 
main(int argc,char ** argv)12ec9796c4SHansol Suh int main(int argc, char **argv)
13ec9796c4SHansol Suh {
14ec9796c4SHansol Suh   /* Initialize TAO and PETSc */
15ec9796c4SHansol Suh   PetscFunctionBeginUser;
16c8025a54SPierre Jolivet   PetscCall(PetscInitialize(&argc, &argv, NULL, help));
17ec9796c4SHansol Suh   PetscCall(RosenbrockMain());
18ec9796c4SHansol Suh   PetscCall(PetscFinalize());
19ec9796c4SHansol Suh   return 0;
20ec9796c4SHansol Suh }
21ec9796c4SHansol Suh 
22ec9796c4SHansol Suh /*TEST
23ec9796c4SHansol Suh 
24ec9796c4SHansol Suh   build:
25ec9796c4SHansol Suh     requires: !complex cuda !single !__float128 !defined(PETSC_USE_65BIT_INDICES)
26ec9796c4SHansol Suh 
27ec9796c4SHansol Suh   test:
28ec9796c4SHansol Suh     suffix: 1
29ec9796c4SHansol Suh     nsize: {{1 2 3}}
30ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_monitor_short -tao_type nls -tao_gatol 1.e-4 -tao_nls_pc_type pbjacobi
31ec9796c4SHansol Suh     requires: !single
32ec9796c4SHansol Suh     output_file: output/rosenbrock1_1.out
33ec9796c4SHansol Suh 
34ec9796c4SHansol Suh   test:
35ec9796c4SHansol Suh     suffix: 2
36ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_monitor_short -tao_type lmvm -tao_gatol 1.e-3
37ec9796c4SHansol Suh     output_file: output/rosenbrock1_2.out
38ec9796c4SHansol Suh 
39ec9796c4SHansol Suh   test:
40ec9796c4SHansol Suh     suffix: 3
41ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_monitor_short -tao_type ntr -tao_gatol 1.e-4
42ec9796c4SHansol Suh     requires: !single
43ec9796c4SHansol Suh     output_file: output/rosenbrock1_3.out
44ec9796c4SHansol Suh 
45ec9796c4SHansol Suh   test:
46ec9796c4SHansol Suh     suffix: 5
47ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_monitor_short -tao_type bntr -tao_gatol 1.e-4
48ec9796c4SHansol Suh     output_file: output/rosenbrock1_5.out
49ec9796c4SHansol Suh 
50ec9796c4SHansol Suh   test:
51ec9796c4SHansol Suh     suffix: 6
52ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_monitor_short -tao_type bntl -tao_gatol 1.e-4
53ec9796c4SHansol Suh     output_file: output/rosenbrock1_6.out
54ec9796c4SHansol Suh 
55ec9796c4SHansol Suh   test:
56ec9796c4SHansol Suh     suffix: 7
57ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_monitor_short -tao_type bnls -tao_gatol 1.e-4
58ec9796c4SHansol Suh     output_file: output/rosenbrock1_7.out
59ec9796c4SHansol Suh 
60ec9796c4SHansol Suh   test:
61ec9796c4SHansol Suh     suffix: 8
62*a336c150SZach Atkins     args: -mat_type aijcusparse -tao_monitor_short -tao_type bntr -tao_bnk_max_cg_its 3 -tao_gatol 1.e-4 -tao_bnk_cg_tao_monitor_short
63ec9796c4SHansol Suh     output_file: output/rosenbrock1_8.out
64ec9796c4SHansol Suh 
65ec9796c4SHansol Suh   test:
66ec9796c4SHansol Suh     suffix: 9
67*a336c150SZach Atkins     args: -mat_type aijcusparse -tao_monitor_short -tao_type bntl -tao_bnk_max_cg_its 3 -tao_gatol 1.e-4 -tao_bnk_cg_tao_monitor_short
68ec9796c4SHansol Suh     output_file: output/rosenbrock1_9.out
69ec9796c4SHansol Suh 
70ec9796c4SHansol Suh   test:
71ec9796c4SHansol Suh     suffix: 10
72*a336c150SZach Atkins     args: -mat_type aijcusparse -tao_monitor_short -tao_type bnls -tao_bnk_max_cg_its 3 -tao_gatol 1.e-4 -tao_bnk_cg_tao_monitor_short
73ec9796c4SHansol Suh     output_file: output/rosenbrock1_10.out
74ec9796c4SHansol Suh 
75ec9796c4SHansol Suh   test:
76ec9796c4SHansol Suh     suffix: 11
77ec9796c4SHansol Suh     args: -mat_type aijcusparse -test_lmvm -tao_type bqnktr -tao_max_it 10 -tao_bqnk_mat_type lmvmbroyden
78ec9796c4SHansol Suh     output_file: output/rosenbrock1_11.out
79ec9796c4SHansol Suh 
80ec9796c4SHansol Suh   test:
81ec9796c4SHansol Suh     suffix: 12
82ec9796c4SHansol Suh     args: -mat_type aijcusparse -test_lmvm -tao_type bqnktr -tao_max_it 10 -tao_bqnk_mat_type lmvmbadbroyden
83ec9796c4SHansol Suh     output_file: output/rosenbrock1_12.out
84ec9796c4SHansol Suh 
85ec9796c4SHansol Suh   test:
86ec9796c4SHansol Suh     suffix: 13
87ec9796c4SHansol Suh     args: -mat_type aijcusparse -test_lmvm -tao_type bqnktr -tao_max_it 10 -tao_bqnk_mat_type lmvmsymbroyden
88ec9796c4SHansol Suh     output_file: output/rosenbrock1_13.out
89ec9796c4SHansol Suh 
90ec9796c4SHansol Suh   test:
91ec9796c4SHansol Suh     suffix: 14
92ec9796c4SHansol Suh     args: -mat_type aijcusparse -test_lmvm -tao_type bqnktr -tao_max_it 10 -tao_bqnk_mat_type lmvmbfgs
93ec9796c4SHansol Suh     output_file: output/rosenbrock1_14.out
94ec9796c4SHansol Suh 
95ec9796c4SHansol Suh   test:
96ec9796c4SHansol Suh     suffix: 15
97ec9796c4SHansol Suh     args: -mat_type aijcusparse -test_lmvm -tao_type bqnktr -tao_max_it 10 -tao_bqnk_mat_type lmvmdfp
98ec9796c4SHansol Suh     output_file: output/rosenbrock1_15.out
99ec9796c4SHansol Suh 
100ec9796c4SHansol Suh   test:
101ec9796c4SHansol Suh     suffix: 16
102ec9796c4SHansol Suh     args: -mat_type aijcusparse -test_lmvm -tao_type bqnktr -tao_max_it 10 -tao_bqnk_mat_type lmvmsr1
103ec9796c4SHansol Suh     output_file: output/rosenbrock1_16.out
104ec9796c4SHansol Suh 
105ec9796c4SHansol Suh   test:
106ec9796c4SHansol Suh     suffix: 17
107ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_monitor_short -tao_gatol 1e-4 -tao_type bqnls
108ec9796c4SHansol Suh     output_file: output/rosenbrock1_17.out
109ec9796c4SHansol Suh 
110ec9796c4SHansol Suh   test:
111ec9796c4SHansol Suh     suffix: 18
112ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_monitor_short -tao_gatol 1e-4 -tao_type blmvm
113ec9796c4SHansol Suh     output_file: output/rosenbrock1_18.out
114ec9796c4SHansol Suh 
115ec9796c4SHansol Suh   test:
116ec9796c4SHansol Suh     suffix: 19
117ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_monitor_short -tao_gatol 1e-4 -tao_type bqnktr -tao_bqnk_mat_type lmvmsr1
118ec9796c4SHansol Suh     output_file: output/rosenbrock1_19.out
119ec9796c4SHansol Suh 
120ec9796c4SHansol Suh   test:
121ec9796c4SHansol Suh     suffix: 20
122ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_monitor -tao_gatol 1e-4 -tao_type blmvm -tao_ls_monitor
123ec9796c4SHansol Suh     output_file: output/rosenbrock1_20.out
124ec9796c4SHansol Suh 
125ec9796c4SHansol Suh   test:
126ec9796c4SHansol Suh     suffix: 21
127ec9796c4SHansol Suh     args: -mat_type aijcusparse -test_lmvm -tao_type bqnktr -tao_max_it 10 -tao_bqnk_mat_type lmvmsymbadbroyden
128ec9796c4SHansol Suh     output_file: output/rosenbrock1_21.out
129ec9796c4SHansol Suh 
130ec9796c4SHansol Suh   test:
131ec9796c4SHansol Suh     suffix: 22
132ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_max_it 1 -tao_converged_reason
133ec9796c4SHansol Suh     output_file: output/rosenbrock1_22.out
134ec9796c4SHansol Suh 
135ec9796c4SHansol Suh   test:
136ec9796c4SHansol Suh     suffix: 23
137ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_max_funcs 0 -tao_converged_reason
138ec9796c4SHansol Suh     output_file: output/rosenbrock1_23.out
139ec9796c4SHansol Suh 
140ec9796c4SHansol Suh   test:
141ec9796c4SHansol Suh     suffix: 24
142ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_gatol 10 -tao_converged_reason
143ec9796c4SHansol Suh     output_file: output/rosenbrock1_24.out
144ec9796c4SHansol Suh 
145ec9796c4SHansol Suh   test:
146ec9796c4SHansol Suh     suffix: 25
147ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_grtol 10 -tao_converged_reason
148ec9796c4SHansol Suh     output_file: output/rosenbrock1_25.out
149ec9796c4SHansol Suh 
150ec9796c4SHansol Suh   test:
151ec9796c4SHansol Suh     suffix: 26
152ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_gttol 10 -tao_converged_reason
153ec9796c4SHansol Suh     output_file: output/rosenbrock1_26.out
154ec9796c4SHansol Suh 
155ec9796c4SHansol Suh   test:
156ec9796c4SHansol Suh     suffix: 27
157ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_steptol 10 -tao_converged_reason
158ec9796c4SHansol Suh     output_file: output/rosenbrock1_27.out
159ec9796c4SHansol Suh 
160ec9796c4SHansol Suh   test:
161ec9796c4SHansol Suh     suffix: 28
162ec9796c4SHansol Suh     args: -mat_type aijcusparse -tao_fmin 10 -tao_converged_reason
163ec9796c4SHansol Suh     output_file: output/rosenbrock1_28.out
164ec9796c4SHansol Suh 
165ec9796c4SHansol Suh   test:
166ec9796c4SHansol Suh     suffix: test_dbfgs
167ec9796c4SHansol Suh     nsize: {{1 2 3}}
168ec9796c4SHansol Suh     output_file: output/rosenbrock1_14.out
1697833cfa3SHansol Suh     args: -mat_type aijcusparse -n 10 -tao_type bqnktr -test_lmvm -tao_max_it 10 -tao_bqnk_mat_type lmvmdbfgs -tao_bqnk_mat_lmvm_scale_type none -tao_bqnk_mat_lbfgs_type {{inplace reorder}} -tao_bqnk_mat_lbfgs_recursive {{0 1}}
170ec9796c4SHansol Suh 
171ec9796c4SHansol Suh   test:
172ec9796c4SHansol Suh     suffix: test_ddfp
173ec9796c4SHansol Suh     nsize: {{1 2 3}}
174ec9796c4SHansol Suh     output_file: output/rosenbrock1_14.out
1757833cfa3SHansol Suh     args: -mat_type aijcusparse -n 10 -tao_type bqnktr -test_lmvm -tao_max_it 10 -tao_bqnk_mat_type lmvmddfp -tao_bqnk_mat_lmvm_scale_type none -tao_bqnk_mat_ldfp_type {{inplace reorder}} -tao_bqnk_mat_ldfp_recursive {{0 1}}
176ec9796c4SHansol Suh 
177ec9796c4SHansol Suh   test:
178ec9796c4SHansol Suh     suffix: test_dqn_1
179ec9796c4SHansol Suh     nsize: 1
180ec9796c4SHansol Suh     output_file: output/rosenbrock1_29.out
1817833cfa3SHansol Suh     args: -mat_type aijcusparse -n 10 -tao_type bqnktr -test_lmvm -tao_max_it 10 -tao_bqnk_mat_type lmvmdqn -tao_bqnk_mat_lmvm_scale_type none -tao_bqnk_mat_lqn_type {{inplace reorder}}
182ec9796c4SHansol Suh 
183ec9796c4SHansol Suh   test:
184ec9796c4SHansol Suh     suffix: test_dqn_2
185ec9796c4SHansol Suh     nsize: 2
186ec9796c4SHansol Suh     output_file: output/rosenbrock1_30.out
1877833cfa3SHansol Suh     args: -mat_type aijcusparse -n 10 -tao_type bqnktr -test_lmvm -tao_max_it 10 -tao_bqnk_mat_type lmvmdqn -tao_bqnk_mat_lmvm_scale_type none -tao_bqnk_mat_lqn_type {{inplace reorder}}
188ec9796c4SHansol Suh 
189ec9796c4SHansol Suh   test:
190ec9796c4SHansol Suh     suffix: test_dqn_3
191ec9796c4SHansol Suh     nsize: 3
192ec9796c4SHansol Suh     output_file: output/rosenbrock1_31.out
1937833cfa3SHansol Suh     args: -mat_type aijcusparse -n 10 -tao_type bqnktr -test_lmvm -tao_max_it 10 -tao_bqnk_mat_type lmvmdqn -tao_bqnk_mat_lmvm_scale_type none -tao_bqnk_mat_lqn_type {{inplace reorder}}
194ec9796c4SHansol Suh 
195ec9796c4SHansol Suh TEST*/
196