Neko 1.99.2
A portable framework for high-order spectral element flow simulations
Loading...
Searching...
No Matches
entropy_viscosity.c
Go to the documentation of this file.
1/*
2 Copyright (c) 2025, The Neko Authors
3 All rights reserved.
4
5 Redistribution and use in source and binary forms, with or without
6 modification, are permitted provided that the following conditions
7 are met:
8
9 * Redistributions of source code must retain the above copyright
10 notice, this list of conditions and the following disclaimer.
11
12 * Redistributions in binary form must reproduce the above
13 copyright notice, this list of conditions and the following
14 disclaimer in the documentation and/or other materials provided
15 with the distribution.
16
17 * Neither the name of the authors nor the names of its
18 contributors may be used to endorse or promote products derived
19 from this software without specific prior written permission.
20
21 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22 "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23 LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
24 FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
25 COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
26 INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
27 BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
28 LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
29 CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
31 ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
32 POSSIBILITY OF SUCH DAMAGE.
33*/
34
35#ifdef __APPLE__
36#include <OpenCL/cl.h>
37#else
38#include <CL/cl.h>
39#endif
40
41#include <stdio.h>
42#include <device/device_config.h>
43#include <device/opencl/jit.h>
44#include <device/opencl/prgm_lib.h>
45#include <device/opencl/check.h>
46
47#include "entropy_viscosity_kernel.cl.h"
48
49void opencl_entropy_visc_compute_residual(void *entropy_residual,
50 void *S, void *S_lag1,
51 void *S_lag2, void *S_lag3,
52 real bdf1, real bdf2,
53 real bdf3, real bdf4,
54 real dt, int n) {
55 cl_int err;
56
57 if (entropy_viscosity_program == NULL)
58 opencl_kernel_jit(entropy_viscosity_kernel,
59 (cl_program *) &entropy_viscosity_program);
60
61 cl_kernel kernel = clCreateKernel(entropy_viscosity_program,
62 "entropy_visc_compute_residual_kernel", &err);
63 CL_CHECK(err);
64
65 CL_CHECK(clSetKernelArg(kernel, 0, sizeof(cl_mem), (void *) &entropy_residual));
66 CL_CHECK(clSetKernelArg(kernel, 1, sizeof(cl_mem), (void *) &S));
67 CL_CHECK(clSetKernelArg(kernel, 2, sizeof(cl_mem), (void *) &S_lag1));
68 CL_CHECK(clSetKernelArg(kernel, 3, sizeof(cl_mem), (void *) &S_lag2));
69 CL_CHECK(clSetKernelArg(kernel, 4, sizeof(cl_mem), (void *) &S_lag3));
70 CL_CHECK(clSetKernelArg(kernel, 5, sizeof(real), &bdf1));
71 CL_CHECK(clSetKernelArg(kernel, 6, sizeof(real), &bdf2));
72 CL_CHECK(clSetKernelArg(kernel, 7, sizeof(real), &bdf3));
73 CL_CHECK(clSetKernelArg(kernel, 8, sizeof(real), &bdf4));
74 CL_CHECK(clSetKernelArg(kernel, 9, sizeof(real), &dt));
75 CL_CHECK(clSetKernelArg(kernel, 10, sizeof(int), &n));
76
77 const int nb = (n + 256 - 1) / 256;
78 const size_t global_item_size = 256 * nb;
79 const size_t local_item_size = 256;
80
81 CL_CHECK(clEnqueueNDRangeKernel((cl_command_queue) glb_cmd_queue, kernel, 1,
82 NULL, &global_item_size, &local_item_size,
83 0, NULL, NULL));
84 CL_CHECK(clReleaseKernel(kernel));
85}
86
87void opencl_entropy_visc_compute_viscosity(void *reg_coeff,
88 void *entropy_residual,
89 void *h,
90 real c_avisc_entropy,
91 real n_S,
92 int n) {
93 cl_int err;
94
95 if (entropy_viscosity_program == NULL)
96 opencl_kernel_jit(entropy_viscosity_kernel,
97 (cl_program *) &entropy_viscosity_program);
98
99 cl_kernel kernel = clCreateKernel(entropy_viscosity_program,
100 "entropy_visc_compute_viscosity_kernel", &err);
101 CL_CHECK(err);
102
103 CL_CHECK(clSetKernelArg(kernel, 0, sizeof(cl_mem), (void *) &reg_coeff));
104 CL_CHECK(clSetKernelArg(kernel, 1, sizeof(cl_mem), (void *) &entropy_residual));
105 CL_CHECK(clSetKernelArg(kernel, 2, sizeof(cl_mem), (void *) &h));
106 CL_CHECK(clSetKernelArg(kernel, 3, sizeof(real), &c_avisc_entropy));
107 CL_CHECK(clSetKernelArg(kernel, 4, sizeof(real), &n_S));
108 CL_CHECK(clSetKernelArg(kernel, 5, sizeof(int), &n));
109
110 const int nb = (n + 256 - 1) / 256;
111 const size_t global_item_size = 256 * nb;
112 const size_t local_item_size = 256;
113
114 CL_CHECK(clEnqueueNDRangeKernel((cl_command_queue) glb_cmd_queue, kernel, 1,
115 NULL, &global_item_size, &local_item_size,
116 0, NULL, NULL));
117 CL_CHECK(clReleaseKernel(kernel));
118}
119
120void opencl_entropy_visc_apply_element_max(void *reg_coeff,
121 int lx,
122 int nelv) {
123 cl_int err;
124 const int lx3 = lx * lx * lx;
125
126 if (entropy_viscosity_program == NULL)
127 opencl_kernel_jit(entropy_viscosity_kernel,
128 (cl_program *) &entropy_viscosity_program);
129
130 cl_kernel kernel = clCreateKernel(entropy_viscosity_program,
131 "entropy_visc_apply_element_max_kernel", &err);
132 CL_CHECK(err);
133
134 CL_CHECK(clSetKernelArg(kernel, 0, sizeof(cl_mem), (void *) &reg_coeff));
135 CL_CHECK(clSetKernelArg(kernel, 1, sizeof(int), &lx3));
136 CL_CHECK(clSetKernelArg(kernel, 2, sizeof(int), &nelv));
137
138 const int local_size = (lx3 < 256) ? lx3 : 256;
139 const size_t global_item_size = local_size * nelv;
140 const size_t local_item_size = local_size;
141
142 CL_CHECK(clEnqueueNDRangeKernel((cl_command_queue) glb_cmd_queue, kernel, 1,
143 NULL, &global_item_size, &local_item_size,
144 0, NULL, NULL));
145 CL_CHECK(clReleaseKernel(kernel));
146}
147
148void opencl_entropy_visc_clamp_to_low_order(void *reg_coeff,
149 void *h,
150 void *max_wave_speed,
151 real c_avisc_low,
152 int n) {
153 cl_int err;
154
155 if (entropy_viscosity_program == NULL)
156 opencl_kernel_jit(entropy_viscosity_kernel,
157 (cl_program *) &entropy_viscosity_program);
158
159 cl_kernel kernel = clCreateKernel(entropy_viscosity_program,
160 "entropy_visc_clamp_to_low_order_kernel", &err);
161 CL_CHECK(err);
162
163 CL_CHECK(clSetKernelArg(kernel, 0, sizeof(cl_mem), (void *) &reg_coeff));
164 CL_CHECK(clSetKernelArg(kernel, 1, sizeof(cl_mem), (void *) &h));
165 CL_CHECK(clSetKernelArg(kernel, 2, sizeof(cl_mem), (void *) &max_wave_speed));
166 CL_CHECK(clSetKernelArg(kernel, 3, sizeof(real), &c_avisc_low));
167 CL_CHECK(clSetKernelArg(kernel, 4, sizeof(int), &n));
168
169 const int nb = (n + 256 - 1) / 256;
170 const size_t global_item_size = 256 * nb;
171 const size_t local_item_size = 256;
172
173 CL_CHECK(clEnqueueNDRangeKernel((cl_command_queue) glb_cmd_queue, kernel, 1,
174 NULL, &global_item_size, &local_item_size,
175 0, NULL, NULL));
176 CL_CHECK(clReleaseKernel(kernel));
177}
178
179void opencl_entropy_visc_smooth_divide(void *reg_coeff,
180 void *temp_field,
181 void *mult_field,
182 int n) {
183 cl_int err;
184
185 if (entropy_viscosity_program == NULL)
186 opencl_kernel_jit(entropy_viscosity_kernel,
187 (cl_program *) &entropy_viscosity_program);
188
189 cl_kernel kernel = clCreateKernel(entropy_viscosity_program,
190 "entropy_visc_smooth_divide_kernel", &err);
191 CL_CHECK(err);
192
193 CL_CHECK(clSetKernelArg(kernel, 0, sizeof(cl_mem), (void *) &reg_coeff));
194 CL_CHECK(clSetKernelArg(kernel, 1, sizeof(cl_mem), (void *) &temp_field));
195 CL_CHECK(clSetKernelArg(kernel, 2, sizeof(cl_mem), (void *) &mult_field));
196 CL_CHECK(clSetKernelArg(kernel, 3, sizeof(int), &n));
197
198 const int nb = (n + 256 - 1) / 256;
199 const size_t global_item_size = 256 * nb;
200 const size_t local_item_size = 256;
201
202 CL_CHECK(clEnqueueNDRangeKernel((cl_command_queue) glb_cmd_queue, kernel, 1,
203 NULL, &global_item_size, &local_item_size,
204 0, NULL, NULL));
205 CL_CHECK(clReleaseKernel(kernel));
206}
207
dt
__global__ void const T *__restrict__ const T *__restrict__ const T *__restrict__ const T *__restrict__ dt
Definition cdtp_kernel.h:109
dirichlet_apply_scalar_kernel
__global__ void dirichlet_apply_scalar_kernel(const int *__restrict__ msk, T *__restrict__ x, const T g, const int m)
Definition dirichlet_kernel.h:42
device_config.h
real
double real
Definition device_config.h:12
opencl_entropy_visc_compute_viscosity
void opencl_entropy_visc_compute_viscosity(void *reg_coeff, void *entropy_residual, void *h, real c_avisc_entropy, real n_S, int n)
Definition entropy_viscosity.c:87
opencl_entropy_visc_compute_residual
void opencl_entropy_visc_compute_residual(void *entropy_residual, void *S, void *S_lag1, void *S_lag2, void *S_lag3, real bdf1, real bdf2, real bdf3, real bdf4, real dt, int n)
Definition entropy_viscosity.c:49
opencl_entropy_visc_smooth_divide
void opencl_entropy_visc_smooth_divide(void *reg_coeff, void *temp_field, void *mult_field, int n)
Definition entropy_viscosity.c:179
opencl_entropy_visc_clamp_to_low_order
void opencl_entropy_visc_clamp_to_low_order(void *reg_coeff, void *h, void *max_wave_speed, real c_avisc_low, int n)
Definition entropy_viscosity.c:148
opencl_entropy_visc_apply_element_max
void opencl_entropy_visc_apply_element_max(void *reg_coeff, int lx, int nelv)
Definition entropy_viscosity.c:120
opencl_kernel_jit
void opencl_kernel_jit(const char *kernel, cl_program *program)
Definition jit.c:50
CL_CHECK
#define CL_CHECK(err)
Definition check.h:12
prgm_lib.h
entropy_viscosity_program
void * entropy_viscosity_program