cuda-practice/Operators.cuh at master · malichan/cuda-practice · GitHub

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
70
71
72
73
74
75
76
77
78
79
80
81
#ifndef Operators_cuh
#define Operators_cuh

#include <limits>

class Operators {
public:
	template<class T>
	class Add {
	public:
		__host__ __device__
		T operator()(T a, T b) const {
			return a + b;
		}

		const T identity;

		Add() : identity(static_cast<T>(0)) {}
	};

	template<class T>
	static Add<T> add() {
		return Add<T>();
	}

	template<class T>
	class Multiply {
	public:
		__host__ __device__
		T operator()(T a, T b) const {
			return a * b;
		}

		const T identity;

		Multiply() : identity(static_cast<T>(1)) {}
	};

	template<class T>
	static Multiply<T> multiply() {
		return Multiply<T>();
	}

	template<class T>
	class Max {
	public:
		__host__ __device__
		T operator()(T a, T b) const {
			return a > b ? a : b;
		}

		const T identity;

		Max() : identity(std::numeric_limits<T>::min()) {}
	};

	template<class T>
	static Max<T> max() {
		return Max<T>();
	}

	template<class T>
	class Min {
	public:
		__host__ __device__
		T operator()(T a, T b) const {
			return a < b ? a : b;
		}

		const T identity;

		Min() : identity(std::numeric_limits<T>::max()) {}
	};

	template<class T>
	static Min<T> min() {
		return Min<T>();
	}
};

#endif