분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
PyTorch 2.14의 @dynamic_spec으로 동적 shape를 compile·export에 한 번에
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
DACON Co.,Ltd | CEO Kookjin Kim | 699-81-01021
Mail-order-sales Registration Number: 2021-서울영등포-1704
Business Providing Employment Information Number: J1204020250004
#901, Eunhaeng-ro 3, Yeongdeungpo-gu, Seoul 07237
E-mail dacon@dacon.io |
Tel. 070-4102-0545
Copyright ⓒ DACON Co.,Ltd All rights reserved