boost/math
2
0
Fork 0
mirror of https://github.com/boostorg/math.git synced 2026-01-19 04:22:09 +00:00
Code Issues Projects Releases Wiki Activity

Add lgamma testing

Browse Source
This commit is contained in:
Matt Borland
2024-08-07 09:37:03 -04:00
parent d4a15eb6b9
commit 0dcf8b5982
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_gamma_nvrtc_double.cpp
View File

@@ -24,12 +24,12 @@ const char* cuda_kernel = R"(
typedef double float_type;
#include <boost/math/special_functions/gamma.hpp>
extern "C" __global__
void test_gamma_kernel(const float_type *in1, const float_type*, float_type *out, int numElements)
void test_gamma_kernel(const float_type *in1, const float_type *in2, float_type *out, int numElements)
{
int i = blockDim.x * blockIdx.x + threadIdx.x;
if (i < numElements)
{
out[i] = boost::math::tgamma(in1[i]);
out[i] = boost::math::tgamma(in1[i]) + boost::math::lgamma(in2[i]);
}
}
)";
@@ -150,7 +150,7 @@ int main()
// Verify Result
for (int i = 0; i < numElements; ++i)
{
auto res = boost::math::tgamma(h_in1[i]);
auto res = boost::math::tgamma(h_in1[i]) + boost::math::lgamma(h_in2[i]);
if (std::isfinite(res))
{
if (boost::math::epsilon_difference(res, h_out[i]) > 300)

6
test/test_gamma_nvrtc_float.cpp
View File

@@ -24,12 +24,12 @@ const char* cuda_kernel = R"(
typedef float float_type;
#include <boost/math/special_functions/gamma.hpp>
extern "C" __global__
void test_gamma_kernel(const float_type *in1, const float_type*, float_type *out, int numElements)
void test_gamma_kernel(const float_type *in1, const float_type *in2, float_type *out, int numElements)
{
int i = blockDim.x * blockIdx.x + threadIdx.x;
if (i < numElements)
{
out[i] = boost::math::tgamma(in1[i]);
out[i] = boost::math::tgamma(in1[i]) + boost::math::lgamma(in2[i]);
}
}
)";
@@ -150,7 +150,7 @@ int main()
// Verify Result
for (int i = 0; i < numElements; ++i)
{
auto res = boost::math::tgamma(h_in1[i]);
auto res = boost::math::tgamma(h_in1[i]) + boost::math::lgamma(h_in2[i]);
if (std::isfinite(res))
{
if (boost::math::epsilon_difference(res, h_out[i]) > 300)