Add missing int64 and uint64 overloads for conj (#1127)

Signed-off-by: Krzysztof Lecki <[email protected]>
SNU-ARC · Feb 7, 2024 · af0a5c8 · af0a5c8
1 parent 50a9095
commit af0a5c8
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/include/cutlass/complex.h b/include/cutlass/complex.h
@@ -462,6 +462,14 @@ CUTLASS_HOST_DEVICE uint32_t conj(uint32_t const& z) {
   return z;
 }
 
+CUTLASS_HOST_DEVICE int64_t conj(int64_t const& z) {
+  return z;
+}
+
+CUTLASS_HOST_DEVICE uint64_t conj(uint64_t const& z) {
+  return z;
+}
+
 CUTLASS_HOST_DEVICE int4b_t conj(int4b_t const& z) {
   return z;
 }