| #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 | |
| } // namespace c10 |