Skip to content

Releases: facebookincubator/AITemplate

AITemplate 0.2

31 Jan 05:49
Compare
Choose a tag to compare
  • FX2AIT: A connector between Pytorch and AITemplate. Check more details in https://github.com/facebookincubator/AITemplate/tree/main/fx2ait/README.md. Another connector between PT2 and AITemplate is WIP, stay tuned.
  • Data type support: Added support for fp32. Bfloat16 is WIP.
  • Profiling: Reduced profiling time for both GeMM and Conv kernels.
  • Tooling: Added simple per-op profiling. Improved error messages.
  • Examples: The StableDiffusion example has been updated.

AITemplate 0.1

11 Oct 07:10
445a20e
Compare
Choose a tag to compare

AITemplate initial release.