markliou's murmur: torch compile for cudagraph

2026年1月17日星期六

torch compile for cudagraph

option 1

TORCH_LOGS="graph_breaks" python transmitter_trainner.py

option 2

import torch._dynamo

# 這會把導致斷圖的原因直接印在螢幕上

torch._dynamo.config.verbose = True

沒有留言:

張貼留言

訂閱：張貼留言 (Atom)