0.2

版本发布时间: 2023-01-31 13:49:01

facebookincubator/AITemplate最新发布版本:0.2(2023-01-31 13:49:01)

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.