PyTorch 테스트 인프라 동작 원리

PyTorch의 테스트는 import 시점에 동적으로 생성되는 특징이 있어, CI(지속적 통합) 환경에서 발생하는 테스트 이름이 소스 코드의 클래스나 메서드 이름과 다를 수 있다.

핵심 동작 원리는 다음과 같다:

• 동적 확장: 데코레이터와 OpInfo(연산자 메타데이터)를 활용해 하나의 테스트 메서드가 여러 디바이스(Device)와 데이터 타입(Dtype) 조합으로 자동 확장된다.
• 디바이스 무관 테스트: instantiate_device_type_tests() 등을 통해 디바이스에 종속되지 않는 테스트를 효율적으로 관리한다.
• CI 샤딩(Sharding): 대규모 테스트를 병렬로 실행하기 위해 테스트를 분할하여 수행한다.

로컬에서 테스트 실패를 재현하려면 pytest -k 옵션이나 test/run_test.py를 사용하는 것이 가장 빠르며, 기여자는 이러한 인프라의 작동 방식을 이해함으로써 더 효과적인 디버깅과 기여가 가능하다.