From 4082fed85a7022a6a79b899796735aa7772994da Mon Sep 17 00:00:00 2001 From: Krzysztof Lecki Date: Fri, 6 Oct 2023 02:01:44 +0200 Subject: [PATCH] Add missing int64 and uint64 overloads for conj (#1127) Signed-off-by: Krzysztof Lecki --- include/cutlass/complex.h | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/include/cutlass/complex.h b/include/cutlass/complex.h index 729d242c..22364d59 100644 --- 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; }