토치 2.0

토치 2.0

torch.compile() 를 사용하면 작업 단위를 전체 그래프로 그룹화하여 작업을 더 빠르게 수행할 수 있습니다.

최적화 예)

  • 전체 그래프를 보면 다음 연산에서 concats output1과 output2의 연산이 있다면 concat은 메모리 값을 옮기지 않고 output1과 output2가 처음 생성될 때 concat 위치에 값을 씁니다.

  • 전체 그래프를 볼 수 있으면 종속성 없이 값에 대해 함수 및 병렬 연산을 수행할 수 있습니다.


출처: https://youtu.be/bwjM20wR3dQ

인용하다