Read the original: PyTorch Blog· Joshua Rosenkranz, Tuan Hoang Trong, Thomas Gooding, Matthew Pisano, and the IBM Spyre Team·Published· 14h agoAI score46
IBM Builds Spyre as a Native PyTorch Device via torch-spyre
Building Spyre as a Native PyTorch Device
AISummary
IBM's torch-spyre integration makes Spyre, its dataflow inference accelerator, a native PyTorch device by mapping PyTorch's device, allocator, stream, and event abstractions onto the Spyre runtime and firmware. Tensors stay resident on device="spyre" between operations, and FX graphs remain in the Inductor compiler path. The approach gives eager and compiled execution one path with lower launch overhead.
Source: PyTorch Blog · pytorch.org