Skip to content
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.

Read the original pytorch.org

Source: PyTorch Blog · pytorch.org