operator <, >, <=, >=, == with n_type

EasternJournalist · Mar 9, 2021 · 88bcafb · 88bcafb
1 parent 0e25e38
commit 88bcafb
Show file tree

Hide file tree

Showing 3 changed files with 120 additions and 0 deletions.
diff --git a/include/DiffVar_arr.h b/include/DiffVar_arr.h
@@ -57,6 +57,26 @@ namespace DiffNum {
 			return value == _Right.value;
 		}
 
+		bool operator <(const n_type _Right) const {
+			return value < _Right;
+		}
+
+		bool operator <=(const n_type _Right) const {
+			return value <= _Right;
+		}
+
+		bool operator >(const n_type _Right) const {
+			return value > _Right;
+		}
+
+		bool operator >=(const n_type _Right) const {
+			return value >= _Right;
+		}
+
+		bool operator ==(const n_type _Right) const {
+			return value == _Right;
+		}
+
 		const d_type& operator[](size_t var_idx) const {
 			return gradient[var_idx];
 		}
@@ -308,6 +328,26 @@ namespace DiffNum {
 			return getValue() == _Right.getValue();
 		}
 
+		bool operator <(const n_type _Right) const {
+			return getValue() < _Right;
+		}
+
+		bool operator <=(const n_type _Right) const {
+			return getValue() <= _Right;
+		}
+
+		bool operator >(const n_type _Right) const {
+			return getValue() > _Right;
+		}
+
+		bool operator >=(const n_type _Right) const {
+			return getValue() >= _Right;
+		}
+
+		bool operator ==(const n_type _Right) const {
+			return getValue() == _Right;
+		}
+
 		const d_type& operator[](size_t var_idx) const {
 			return gradient[var_idx];
 		}

diff --git a/include/DiffVar_cuda.h b/include/DiffVar_cuda.h
@@ -75,6 +75,26 @@ namespace DiffNum {
 			return value == _Right.value;
 		}
 
+		__host__ __device__ bool operator <(const n_type _Right) const {
+			return value < _Right;
+		}
+
+		__host__ __device__ bool operator <=(const n_type _Right) const {
+			return value <= _Right;
+		}
+
+		__host__ __device__ bool operator >(const n_type _Right) const {
+			return value > _Right;
+		}
+
+		__host__ __device__ bool operator >=(const n_type _Right) const {
+			return value >= _Right;
+		}
+
+		__host__ __device__ bool operator ==(const n_type _Right) const {
+			return value == _Right;
+		}
+
 		__host__ __device__ const d_type& operator[](size_t var_idx) const {
 			return gradient[var_idx];
 		}
@@ -326,6 +346,26 @@ namespace DiffNum {
 			return gradient[var_idx];
 		}
 
+		__host__ __device__ bool operator <(const n_type _Right) const {
+			return getValue() < _Right;
+		}
+
+		__host__ __device__ bool operator <=(const n_type _Right) const {
+			return getValue() <= _Right;
+		}
+
+		__host__ __device__ bool operator >(const n_type _Right) const {
+			return getValue() > _Right;
+		}
+
+		__host__ __device__ bool operator >=(const n_type _Right) const {
+			return getValue() >= _Right;
+		}
+
+		__host__ __device__ bool operator ==(const n_type _Right) const {
+			return getValue() == _Right;
+		}
+
 		__host__ __device__ DiffVar_cuda() {}
 
 		__host__ __device__ DiffVar_cuda(const n_type value) : value(value) {

diff --git a/include/DiffVar_vec.h b/include/DiffVar_vec.h
@@ -50,6 +50,26 @@ namespace DiffNum {
 			return value == _Right.value;
 		}
 
+		bool operator <(const n_type _Right) const {
+			return value < _Right;
+		}
+
+		bool operator <=(const n_type _Right) const {
+			return value <= _Right;
+		}
+
+		bool operator >(const n_type _Right) const {
+			return value > _Right;
+		}
+
+		bool operator >=(const n_type _Right) const {
+			return value >= _Right;
+		}
+
+		bool operator ==(const n_type _Right) const {
+			return value == _Right;
+		}
+
 		const d_type& operator[](size_t var_idx) const {
 			return gradient[var_idx];
 		}
@@ -339,6 +359,26 @@ namespace DiffNum {
 			return getValue() == _Right.getValue();
 		}
 
+		bool operator <(const n_type _Right) const {
+			return getValue() < _Right;
+		}
+
+		bool operator <=(const n_type _Right) const {
+			return getValue() <= _Right;
+		}
+
+		bool operator >(const n_type _Right) const {
+			return getValue() > _Right;
+		}
+
+		bool operator >=(const n_type _Right) const {
+			return getValue() >= _Right;
+		}
+
+		bool operator ==(const n_type _Right) const {
+			return getValue() == _Right;
+		}
+
 		const d_type& operator[](size_t var_idx) const {
 			return gradient[var_idx];
 		}