mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
968b94bef2
This patch enables clang-tidy coverage on c10/**/*.h and contains other fixes. Pull Request resolved: https://github.com/pytorch/pytorch/pull/116082 Approved by: https://github.com/Skylion007
22 lines
564 B
C++
22 lines
564 B
C++
#pragma once
|
|
|
|
#include <cstddef>
|
|
|
|
namespace c10 {
|
|
|
|
#ifdef C10_MOBILE
|
|
// Use 16-byte alignment on mobile
|
|
// - ARM NEON AArch32 and AArch64
|
|
// - x86[-64] < AVX
|
|
constexpr size_t gAlignment = 16;
|
|
#else
|
|
// Use 64-byte alignment should be enough for computation up to AVX512.
|
|
constexpr size_t gAlignment = 64;
|
|
#endif
|
|
|
|
constexpr size_t gPagesize = 4096;
|
|
// since the default thp pagesize is 2MB, enable thp only
|
|
// for buffers of size 2MB or larger to avoid memory bloating
|
|
constexpr size_t gAlloc_threshold_thp = static_cast<size_t>(2) * 1024 * 1024;
|
|
} // namespace c10
|