StableHLO
StableHLO
PyTorch나 JAX에서 생성된 연산 그래프를 컴파일러가 처리할 수 있도록 표현하는 중간 표현입니다. TorchTPU에서는 PyTorch 연산이 StableHLO로 낮아진 뒤 XLA를 거쳐 TPU 실행 파일로 변환됩니다. 따라서 사용자 인터페이스는 PyTorch에 가깝게 유지하면서 TPU용 컴파일 경로를 활용할 수 있습니다.
StableHLO
PyTorch나 JAX에서 생성된 연산 그래프를 컴파일러가 처리할 수 있도록 표현하는 중간 표현입니다. TorchTPU에서는 PyTorch 연산이 StableHLO로 낮아진 뒤 XLA를 거쳐 TPU 실행 파일로 변환됩니다. 따라서 사용자 인터페이스는 PyTorch에 가깝게 유지하면서 TPU용 컴파일 경로를 활용할 수 있습니다.