del-geo-cpp-headers 0.1.33

CUDA device utility functions
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

#pragma once
#include <cuda/std/array>

namespace vec3 {

__device__
auto norm(const float* a) -> float
{
   const float sqnrm = a[0] * a[0] + a[1] * a[1] + a[2] * a[2];
   return sqrt(sqnrm);
}

__device__
auto add(const float* a, const float* b) -> cuda::std::array<float,3>
{
    return {a[0] + b[0], a[1] + b[1], a[2] + b[2]};
}

__device__
auto sub(const float* a, const float* b) -> cuda::std::array<float,3>
{
    return {a[0] - b[0], a[1] - b[1], a[2] - b[2]};
}

__device__
auto cross(const float* v1, const float* v2) -> cuda::std::array<float,3>
{
    return {
        v1[1] * v2[2] - v2[1] * v1[2],
        v1[2] * v2[0] - v2[2] * v1[0],
        v1[0] * v2[1] - v2[0] * v1[1],
    };
}

__device__
auto dot(const float* a, const float* b) -> float
{
    return a[0] * b[0] + a[1] * b[1] + a[2] * b[2];
}

__device__
auto axpy(float a, const float* x, const float* y) -> cuda::std::array<float,3>
{
    return {
        a * x[0] + y[0],
        a * x[1] + y[1],
        a * x[2] + y[2],
    };
}

__device__
auto scale(const float* x, float a) -> cuda::std::array<float,3>
{
    return {
        a * x[0],
        a * x[1],
        a * x[2],
    };
}

__device__
void add_inplace(float* x, const float* y) {
    x[0] += y[0];
    x[1] += y[1];
    x[2] += y[2];
}


}