From a31b43b3f3cb609c6ed0b30ca330d766291e84fb Mon Sep 17 00:00:00 2001 From: psaab Date: Thu, 2 Mar 2023 08:17:21 -0800 Subject: [PATCH] Re-enable aarch64 support lost in 277bd6e5379e0c1e1eb64db1a654b30e1efddc8e (#846) --- include/cutlass/uint128.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/include/cutlass/uint128.h b/include/cutlass/uint128.h index 38d5b4d5..c3e3275d 100644 --- a/include/cutlass/uint128.h +++ b/include/cutlass/uint128.h @@ -54,7 +54,7 @@ namespace cutlass { ///////////////////////////////////////////////////////////////////////////////////////////////// /// Optionally enable GCC's built-in type -#if defined(__x86_64) && !defined(__CUDA_ARCH__) && defined(__GNUC__) +#if (defined(__x86_64) || defined (__aarch64__)) && !defined(__CUDA_ARCH__) && defined(__GNUC__) #define CUTLASS_UINT128_NATIVE #elif defined(_MSC_VER) && defined(_M_AMD64) && !defined(__CUDA_ARCH__) #define CUTLASS_INT128_ARITHMETIC