[cpython] CPython arraymodule 최적화: 구조체 메모리 레이아웃 개선을 통한 성능 향상

2026년 5월 6일수정: 2026년 5월 6일

PR 링크: python/cpython#149455 상태: Merged | 변경: +None / -None

들어가며

CPython의 arraymodule은 파이썬의 array 모듈을 구현하는 핵심 컴포넌트입니다. 최근 CPython 레포지토리에 병합된 gh-148675는 arraydescr 구조체의 메모리 레이아웃을 개선하여 성능을 최적화했습니다. 이 변경은 단순히 코드 스타일을 바꾸는 것이 아니라, 구조체의 크기를 줄이고 불필요한 메모리 참조(indirection)를 제거하여 CPU 캐시 효율성을 높이는 데 목적이 있습니다.

코드 분석

1. 구조체 정의 변경 (`Modules/arraymodule.c`)

가장 핵심적인 변경은 arraydescr 구조체 내부의 typecode 필드 타입 변경입니다.

Before:

struct arraydescr {
    const char *typecode;
    int itemsize;
    // ...
};

After:

struct arraydescr {
    char typecode[3];  // big enough to store "Zd\0"
    int itemsize;
    // ...
};

기존에는 const char *를 사용하여 문자열 리터럴을 가리키는 포인터를 저장했습니다. 이는 8바이트(64비트 환경 기준)의 포인터 크기를 차지하며, 실제 문자열 데이터에 접근하기 위해 추가적인 메모리 참조가 필요했습니다. 이를 char[3]으로 변경함으로써 구조체 내부에 데이터를 직접 포함(inline)시켰습니다. 이는 메모리 단편화를 줄이고 CPU가 데이터를 더 빠르게 읽어올 수 있게 합니다.

2. 센티넬(Sentinel) 및 루프 조건 변경

구조체가 포인터에서 배열로 바뀌었으므로, 리스트의 끝을 알리는 센티넬 체크 방식도 변경되었습니다.

Before:

for (descr = descriptors; descr->typecode != NULL; descr++) {
    if (strcmp(descr->typecode, typecode) == 0) { ... }
}

After:

for (descr = descriptors; descr->typecode[0] != 0; descr++) {
    if (strcmp(descr->typecode, typecode) == 0) { ... }
}

typecode가 포인터일 때는 NULL 체크를 했지만, 이제는 배열의 첫 번째 바이트가 0인지 확인하는 방식으로 변경되었습니다. 또한, descriptors 배열의 마지막 요소도 NULL 대신 빈 문자열 ""로 초기화하여 일관성을 유지했습니다.

왜 이게 좋은가

메모리 참조 감소 (Avoid Indirection): 포인터를 따라가서 문자열을 읽는 과정이 사라지고, 구조체 메모리 블록 내에서 즉시 데이터를 읽을 수 있습니다. 이는 CPU 캐시 미스(Cache Miss) 확률을 낮춥니다.
구조체 크기 최적화: 포인터 크기(8바이트)를 3바이트로 줄여 구조체 전체 크기를 최적화했습니다. 이는 descriptors 배열이 메모리에 로드될 때 더 적은 공간을 차지하게 합니다.
컴파일러 최적화 가능성: 리뷰어 scoder가 언급했듯이, strcmp 호출을 단일 문자 비교로 최적화할 여지가 생겼습니다. 현재는 strcmp를 유지하고 있지만, 향후 컴파일러가 더 공격적으로 최적화할 수 있는 구조가 되었습니다.