PyTorch 2.14의 @dynamic_spec으로 동적 shape를 compile·export에 한 번에

2026.09.12 17:18 55 조회

PyTorch 2.14는 @dynamic_spec·ShapesSpec으로 가변 차원을 함수·모듈에 한 번 선언하면 torch.compile, torch.export, make_fx가 같은 스펙을 읽습니다. 호출마다 다른 방식으로 동적 shape를 넘기던 조건을 줄일 수 있습니다.

자세한 내용과 근거는 DAKER 원문에서 확인하실 수 있습니다.


핵심 요약

• @dynamic_spec으로 가변 차원을 한 번 선언하면 compile·export가 같은 스펙을 읽습니다.

• 호출마다 다른 동적 shape 조건을 줄일 수 있습니다.

• PyTorch 2.14 공식 릴리스 안내를 기준으로 정리했습니다.


원문 보기: https://daker.ai/community/pytorch-2-14-dynamic-spec-shapes-compile-export