Cast return value to int64_t for cache size (#28814)

Signed-off-by: tiehexue <tiehexue@hotmail.com>
This commit is contained in:
tiehexue 2025-11-18 00:02:32 +08:00 committed by GitHub
parent 7f064491f8
commit e42bd8c2e3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -754,7 +754,7 @@ class AttentionScheduler {
return l2_cache_size >> 1; // use 50% of L2 cache return l2_cache_size >> 1; // use 50% of L2 cache
} }
// Fallback if sysctlbyname fails // Fallback if sysctlbyname fails
return 128 * 1024 >> 1; // use 50% of 128KB return 128LL * 1024 >> 1; // use 50% of 128KB
#else #else
long l2_cache_size = sysconf(_SC_LEVEL2_CACHE_SIZE); long l2_cache_size = sysconf(_SC_LEVEL2_CACHE_SIZE);
TORCH_CHECK_NE(l2_cache_size, -1); TORCH_CHECK_NE(l2_cache_size, -1);