PyTorch-Plugin-FL 概览

PyTorch-Plugin-FL 概览#

torch_fl 是一个基于 PrivateUse1 扩展机制的自定义 PyTorch 设备插件，将 FlagGems 高性能 Triton 算子注册为 flagos 设备后端，以实现统一的多芯片支持。