|
1 | | -// Solve the Momentum equation |
2 | | -#ifdef GPUSolver_ |
3 | | - start1 = std::clock(); |
4 | | - int offset = 0; |
5 | | - const tmp<volScalarField> nuEff_tmp(turbulence->nuEff()); |
6 | | - const volScalarField& nuEff = nuEff_tmp(); |
7 | | - forAll(U.boundaryField(), patchi) |
8 | | - { |
9 | | - const scalarField& patchP = p.boundaryField()[patchi]; |
10 | | - const vectorField& patchU = U.boundaryField()[patchi]; |
11 | | - const scalarField& patchRho = rho.boundaryField()[patchi]; |
12 | | - const scalarField& patchNuEff = nuEff.boundaryField()[patchi]; |
13 | | - |
14 | | - int patchSize = patchP.size(); |
15 | | - |
16 | | - // boundary pressure |
17 | | - memcpy(boundary_pressure_init+offset, &patchP[0], patchSize*sizeof(double)); |
18 | | - // boundary velocity |
19 | | - memcpy(boundary_velocity_init+3*offset, &patchU[0][0], 3*patchSize*sizeof(double)); |
20 | | - // boundary nuEff |
21 | | - memcpy(boundary_nuEff_init+offset, &patchNuEff[0], patchSize*sizeof(double)); |
22 | | - // boundary rho |
23 | | - memcpy(boundary_rho_init+offset, &patchRho[0], patchSize*sizeof(double)); |
24 | | - offset += patchSize; |
25 | | - } |
26 | | - end1 = std::clock(); |
27 | | - time_monitor_UEqn += double(end1 - start1) / double(CLOCKS_PER_SEC); |
28 | | - time_monitor_UEqn_mtxAssembly += double(end1 - start1) / double(CLOCKS_PER_SEC); |
29 | | - time_monitor_UEqn_mtxAssembly_CPU_prepare += double(end1 - start1) / double(CLOCKS_PER_SEC); |
30 | | - |
31 | | - start1 = std::clock(); |
32 | | - UEqn_GPU.initializeTimeStep(); |
33 | | - U.oldTime(); |
34 | | - UEqn_GPU.fvm_ddt(&U.oldTime()[0][0]); |
35 | | - UEqn_GPU.fvm_div(boundary_pressure_init, boundary_velocity_init, boundary_nuEff_init, boundary_rho_init); |
36 | | - UEqn_GPU.fvc_grad(&p[0]); |
37 | | - UEqn_GPU.fvc_grad_vector(); |
38 | | - UEqn_GPU.dev2T(); |
39 | | - UEqn_GPU.fvc_div_tensor(&nuEff[0]); |
40 | | - UEqn_GPU.fvm_laplacian(); |
41 | | - UEqn_GPU.sync(); |
42 | | - end1 = std::clock(); |
43 | | - time_monitor_UEqn += double(end1 - start1) / double(CLOCKS_PER_SEC); |
44 | | - time_monitor_UEqn_mtxAssembly += double(end1 - start1) / double(CLOCKS_PER_SEC); |
45 | | - time_monitor_UEqn_mtxAssembly_GPU_run += double(end1 - start1) / double(CLOCKS_PER_SEC); |
46 | | - |
47 | | - // start2 = std::clock(); |
48 | | - // fvVectorMatrix turb_source |
49 | | - // ( |
50 | | - // turbulence->divDevRhoReff(U) |
51 | | - // ); |
52 | | - // end2 = std::clock(); |
53 | | - // time_monitor_CPU += double(end2 - start2) / double(CLOCKS_PER_SEC); |
54 | | - |
55 | | - // UEqn_GPU.add_fvMatrix(&turb_source.lower()[0], &turb_source.diag()[0], &turb_source.upper()[0], &turb_source.source()[0][0]); |
56 | | - // end1 = std::clock(); |
57 | | - // time_monitor_UEqn += double(end1 - start1) / double(CLOCKS_PER_SEC); |
58 | | - // time_monitor_UEqn_mtxAssembly += double(end1 - start1) / double(CLOCKS_PER_SEC); |
59 | | - |
60 | | - // check value |
61 | | - // U.oldTime(); |
62 | | - // tmp<fvVectorMatrix> tUEqn |
63 | | - // ( |
64 | | - // fvm::ddt(rho, U) |
65 | | - // + |
66 | | - // fvm::div(phi, U) |
67 | | - // + |
68 | | - // turbulence->divDevRhoReff(U) |
69 | | - // == -fvc::grad(p) |
70 | | - // ); |
71 | | - // fvVectorMatrix& UEqn = tUEqn.ref(); |
72 | | - // printf("b_cpu = %e\n", UEqn.source()[1][1]); |
73 | | - // forAll(U.boundaryField(), patchi){ |
74 | | - // labelUList sub_boundary = mesh.boundary()[patchi].faceCells(); |
75 | | - // forAll(sub_boundary, i){ |
76 | | - // if (sub_boundary[i] == 1){ |
77 | | - // printf("b_cpu_bou = %e\n", UEqn.boundaryCoeffs()[patchi][i][1]); |
78 | | - // printf("patchi = %d, i = %d\n", patchi, i); |
79 | | - // } |
80 | | - // } |
81 | | - // } |
82 | | - // if (pimple.momentumPredictor()) |
83 | | - // { |
84 | | - // solve(UEqn); |
85 | | - // Info << "U_CPU\n" << U << endl; |
86 | | - // K = 0.5*magSqr(U); |
87 | | - // } |
88 | | - // UEqn_GPU.checkValue(true); |
89 | | -#else |
90 | | - start1 = std::clock(); |
91 | | - tmp<fvVectorMatrix> tUEqn |
92 | | - ( |
93 | | - fvm::ddt(rho, U) + fvm::div(phi, U) |
94 | | - + turbulence->divDevRhoReff(U) |
95 | | - == -fvc::grad(p) |
96 | | - ); |
97 | | - fvVectorMatrix& UEqn = tUEqn.ref(); |
98 | | - |
99 | | - end1 = std::clock(); |
100 | | - time_monitor_UEqn += double(end1 - start1) / double(CLOCKS_PER_SEC); |
101 | | - time_monitor_UEqn_mtxAssembly += double(end1 - start1) / double(CLOCKS_PER_SEC); |
102 | | - |
103 | | - UEqn.relax(); |
104 | | - start1 = std::clock(); |
105 | | - if (pimple.momentumPredictor()) |
106 | | - { |
107 | | - solve(UEqn); |
108 | | - |
109 | | - K = 0.5*magSqr(U); |
110 | | - } |
111 | | - end1 = std::clock(); |
112 | | - time_monitor_UEqn += double(end1 - start1) / double(CLOCKS_PER_SEC); |
113 | | - time_monitor_UEqn_solve += double(end1 - start1) / double(CLOCKS_PER_SEC); |
114 | | -#endif |
115 | | - |
116 | | -// start1 = std::clock(); |
117 | | -// // // std::thread t(&dfMatrix::solve, &UEqn_GPU); |
118 | | -// UEqn_GPU.solve(); |
119 | | -// end1 = std::clock(); |
120 | | -// time_monitor_UEqn += double(end1 - start1) / double(CLOCKS_PER_SEC); |
121 | | -// time_monitor_UEqn_solve += double(end1 - start1) / double(CLOCKS_PER_SEC); |
122 | | - |
123 | | -// start1 = std::clock(); |
124 | | -// // // t.join(); |
125 | | -// // UEqn_GPU.updatePsi(&U[0][0]); |
126 | | -// K = 0.5*magSqr(U); |
127 | | -// end1 = std::clock(); |
128 | | -// time_monitor_UEqn += double(end1 - start1) / double(CLOCKS_PER_SEC); |
129 | | -// time_monitor_UEqn_mtxAssembly += double(end1 - start1) / double(CLOCKS_PER_SEC); |
130 | | -// time_monitor_CPU += double(end1 - start1) / double(CLOCKS_PER_SEC); |
131 | | -// // Info << "U_amgx = " << U << endl; |
| 1 | +start1 = std::clock(); |
| 2 | +tmp<fvVectorMatrix> tUEqn |
| 3 | +( |
| 4 | + fvm::ddt(rho, U) + fvm::div(phi, U) |
| 5 | + + turbulence->divDevRhoReff(U) |
| 6 | +); |
| 7 | +fvVectorMatrix& UEqn = tUEqn.ref(); |
| 8 | + |
| 9 | +end1 = std::clock(); |
| 10 | +time_monitor_UEqn += double(end1 - start1) / double(CLOCKS_PER_SEC); |
| 11 | +time_monitor_UEqn_mtxAssembly += double(end1 - start1) / double(CLOCKS_PER_SEC); |
| 12 | + |
| 13 | +UEqn.relax(); |
| 14 | +start1 = std::clock(); |
| 15 | +if (pimple.momentumPredictor()) |
| 16 | +{ |
| 17 | + solve(UEqn == -fvc::grad(p)); |
| 18 | + |
| 19 | + K = 0.5*magSqr(U); |
| 20 | +} |
| 21 | +end1 = std::clock(); |
| 22 | +time_monitor_UEqn += double(end1 - start1) / double(CLOCKS_PER_SEC); |
| 23 | +time_monitor_UEqn_solve += double(end1 - start1) / double(CLOCKS_PER_SEC); |
132 | 24 |
|
0 commit comments