Tagged: torch.compile

2 articles

Latest

PyTorch's nn.Linear already runs a fused kernel—here's why torch.compile adds nothing for one layer

PyTorch profiling isn't just for experts: Here's how to read your first trace