3. 功能支持

mcApex以兼容Apex的原生使用方式为设计目标,当前支持部分Apex的接口。

本章参照Apex官方文档内容,介绍mcApex各模块使用的兼容性和注意事项。其中:

  • 支持:支持并兼容原生用法

  • 部分支持:部分支持原生用法,或API行为和原生用法有区别

3.1. apex.fused_dense.FusedDense

支持

3.2. apex.transformer.functional.FusedScaleMaskSoftmax

支持

3.3. apex.transformer.functional.fused_apply_rotary_pos_emb

支持

3.4. apex.transformer.functional.fused_apply_rotary_pos_emb_cached

支持

3.5. apex.transformer.functional.fused_apply_rotary_pos_emb_thd

支持

3.6. apex.normalization.FusedLayerNorm

支持

3.7. apex.normalization.FusedRMSNorm

支持

3.8. apex.normalization.MixedFusedLayerNorm

支持

3.9. apex.normalization.MixedFusedRMSNorm

支持

3.10. apex.contrib.layer_norm.FastLayerNorm

支持