Skip to content

Commit 7e4c591

Browse files
committed
Fix kernel inclusion
1 parent 71653a3 commit 7e4c591

2 files changed

Lines changed: 4 additions & 0 deletions

File tree

stan/math/opencl/kernels/ordered_logistic_glm_lpmf.hpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
#include <stan/math/opencl/kernel_cl.hpp>
66
#include <stan/math/opencl/kernels/device_functions/log1m_exp.hpp>
77
#include <stan/math/opencl/kernels/device_functions/log1p_exp.hpp>
8+
#include <stan/math/opencl/kernels/device_functions/inv_logit.hpp>
89

910
namespace stan {
1011
namespace math {
@@ -171,6 +172,7 @@ const kernel_cl<out_buffer, out_buffer, out_buffer, out_buffer, in_buffer,
171172
in_buffer, in_buffer, in_buffer, int, int, int, int, int, int>
172173
ordered_logistic_glm("ordered_logistic_glm",
173174
{log1p_exp_device_function, log1m_exp_device_function,
175+
inv_logit_device_function,
174176
ordered_logistic_glm_kernel_code},
175177
{{"REDUCTION_STEP_SIZE", 4}, {"LOCAL_SIZE_", 64}});
176178

stan/math/opencl/kernels/ordered_logistic_lpmf.hpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
#include <stan/math/opencl/kernel_cl.hpp>
66
#include <stan/math/opencl/kernels/device_functions/log1m_exp.hpp>
77
#include <stan/math/opencl/kernels/device_functions/log1p_exp.hpp>
8+
#include <stan/math/opencl/kernels/device_functions/inv_logit.hpp>
89

910
namespace stan {
1011
namespace math {
@@ -165,6 +166,7 @@ const kernel_cl<out_buffer, out_buffer, out_buffer, in_buffer, in_buffer,
165166
in_buffer, int, int, int, int, int, int>
166167
ordered_logistic("ordered_logistic",
167168
{log1p_exp_device_function, log1m_exp_device_function,
169+
inv_logit_device_function,
168170
ordered_logistic_kernel_code},
169171
{{"REDUCTION_STEP_SIZE", 4}, {"LOCAL_SIZE_", 64}});
170172

0 commit comments

Comments
 (0)