From d49bef88f9e377dca83781328a987e49fee723b8 Mon Sep 17 00:00:00 2001 From: psaab Date: Fri, 20 Jan 2023 12:51:58 -0800 Subject: [PATCH] Enable aarch64 support (#779) --- 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 116b9502..2717e022 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