diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake index 6bd9dcbdb984dc502acfbba157e7a12f650d0248..7d1995360b47416fdd59c759147dd239d1157a04 100644 --- a/cmake/cuda.cmake +++ b/cmake/cuda.cmake @@ -4,6 +4,7 @@ endif() if (WITH_NV_JETSON) + add_definitions(-DWITH_NV_JETSON) set(paddle_known_gpu_archs "53 62 72") set(paddle_known_gpu_archs7 "53") set(paddle_known_gpu_archs8 "53 62") diff --git a/paddle/fluid/platform/cpu_info.cc b/paddle/fluid/platform/cpu_info.cc index 6b9bacd7b71b0881660b68c7c6248848ac9f34fc..6545e62927fe05b1d7a3426b3d91c5eb6aac0c2d 100644 --- a/paddle/fluid/platform/cpu_info.cc +++ b/paddle/fluid/platform/cpu_info.cc @@ -139,6 +139,7 @@ bool MayIUse(const cpu_isa_t cpu_isa) { if (cpu_isa == isa_any) { return true; } else { +#ifndef WITH_NV_JETSON int reg[4]; cpuid(reg, 0); int nIds = reg[0]; @@ -164,6 +165,7 @@ bool MayIUse(const cpu_isa_t cpu_isa) { return (reg[1] & avx512f_mask) != 0; } } +#endif return false; } } diff --git a/paddle/fluid/platform/cpu_info.h b/paddle/fluid/platform/cpu_info.h index 3255f50d4a20de6505cbdbe32a485dc69d31c62e..66f05d51c0b6795e7509b3e80d9ffd3e4a28adfd 100644 --- a/paddle/fluid/platform/cpu_info.h +++ b/paddle/fluid/platform/cpu_info.h @@ -40,12 +40,14 @@ limitations under the License. */ #ifdef _WIN32 #define cpuid(reg, x) __cpuidex(reg, x, 0) #else +#ifndef WITH_NV_JETSON #include inline void cpuid(int reg[4], int x) { __cpuid_count(x, 0, reg[0], reg[1], reg[2], reg[3]); } #endif #endif +#endif namespace paddle { namespace platform {