PyTorch-Plugin-FL Overview

torch_fl is a custom PyTorch device plugin based on the PrivateUse1 extension mechanism, registering FlagGems high-performance Triton operators as the flagos device backend for unified multi-chip support.

• Features
  • Automatic device registration
  • Configurable backend routing
  • Multi-platform support
  • Complete device management API
• Architecture overview
• Project Structure