Enable aarch64 support (#779)

NVIDIA · Jan 20, 2023 · d49bef8 · d49bef8
1 parent 8b42e75
commit d49bef8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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