Open
feat: support float8 e8m0 scale nn transdata #1
jayshuAdd on Aug 11
feat: support float8 e8m0 scale nn transdata #1
Open
jayshuAdd on Aug 11
jayshu
jayshuCollaborators
Aug 11

Description

Support FLOAT8_E8M0 scale ND to Nn storage conversion through aclnnNpuFormatCast when dstFormat is FRACTAL_NZ. The Nn storage shape is (..., ceil(N / 16), KScale, 16, 2), while the original logical shape remains (..., KScale, N, 2).

Changes

  • Add shape calculation and validation for FLOAT8_E8M0 scale ND to Nn.
  • Add Reshape + Transpose based transdata implementation.
  • Update aclnnNpuFormatCast API docs.
  • Add Ascend950 L2 unit cases for valid 3D/4D shapes and invalid N alignment.

Test

  • git diff --check for modified files

Type

likedislike
Merge blocked
jayshujayshuCollaborators
Aug 15 Pushed 1 commits:3d3e2f03-fix npu format cast e8m0 nn aclnn path
jayshujayshuCollaborators
Aug 17 Force-pushed 1 commits:cf442912-fix npu format cast e8m0 nn aclnn path
jayshujayshuCollaborators
Aug 17 Force-pushed 1 commits:f0b059e5-fix npu format cast e8m0 nn aclnn path
jayshujayshuCollaborators
Aug 17 Force-pushed 1 commits:7319a71b-fix npu format cast e8m0 nn aclnn path
jayshujayshuCollaborators
Aug 25 Pushed 1 commits:72208bfd-docs: describe e8m0 format cast as affinity format
jayshujayshuCollaborators
Aug 25 Pushed commit
jayshujayshuCollaborators
Aug 25 Pushed 1 commits:ddb711cf-docs: refine e8m0 scale nz format cast
jayshujayshuCollaborators
Aug 26 Pushed 1 commits:3e711737-style: format npu format cast changes