From 5b6d62a9a94ede4d3f4b8d47fb63af04f46e2b51 Mon Sep 17 00:00:00 2001 From: Andrew Guschin Date: Wed, 30 Apr 2025 15:43:07 +0400 Subject: [PATCH] Add detection of loongarch64 CPUs --- src/common/sdk/nvidia/inc/cpuopsys.h | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/common/sdk/nvidia/inc/cpuopsys.h b/src/common/sdk/nvidia/inc/cpuopsys.h index 230c7b5214..68c7928128 100644 --- a/src/common/sdk/nvidia/inc/cpuopsys.h +++ b/src/common/sdk/nvidia/inc/cpuopsys.h @@ -181,6 +181,10 @@ # endif #endif +#if defined(__loongarch__) && (__loongarch_grlen == 64) +# define NVCPU_LOONGARCH64 +#endif + #if defined(__arm__) || defined(_M_ARM) /* * 32-bit instruction set on, e.g., ARMv7 or AArch32 execution state @@ -232,7 +236,8 @@ #if defined(NVCPU_IA64) || defined(NVCPU_X86_64) || \ defined(NV_MACINTOSH_64) || defined(NVCPU_AARCH64) || \ - defined(NVCPU_PPC64LE) || defined(NVCPU_RISCV64) + defined(NVCPU_PPC64LE) || defined(NVCPU_RISCV64) || \ + defined(NVCPU_LOONGARCH64) # define NV_64_BITS /* all architectures where pointers are 64 bits */ #else /* we assume 32 bits. I don't see a need for NV_16_BITS. */