Const-qualify half_t::operator+/* (#1726)

These arithmetic operations do not and should not modify the current instance. Enables adding and multiplying const-qualified values of type half_t.
NVIDIA · May 13, 2024 · 4640fae · 4640fae
1 parent 12c2892
commit 4640fae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cub/test/half.h b/cub/test/half.h
@@ -238,7 +238,7 @@ struct half_t
   }
 
   /// Multiply
-  __host__ __device__ __forceinline__ half_t operator*(const half_t& other)
+  __host__ __device__ __forceinline__ half_t operator*(const half_t& other) const
   {
     return half_t(float(*this) * float(other));
   }
@@ -250,7 +250,7 @@ struct half_t
   }
 
   /// Add
-  __host__ __device__ __forceinline__ half_t operator+(const half_t& other)
+  __host__ __device__ __forceinline__ half_t operator+(const half_t& other) const
   {
     return half_t(float(*this) + float(other));
   }