Skip to content

Rename ARM FP8 inline helpers to NEON naming conventions#426

Merged
JakeStevens merged 1 commit into
pytorch:mainfrom
fbarchard:neon_fp8
Jul 23, 2026
Merged

Rename ARM FP8 inline helpers to NEON naming conventions#426
JakeStevens merged 1 commit into
pytorch:mainfrom
fbarchard:neon_fp8

Conversation

@fbarchard

@fbarchard fbarchard commented Jul 23, 2026

Copy link
Copy Markdown
Contributor

Rename cpuinfo_has_arm_f8dot and cpuinfo_has_arm_f8mm to cpuinfo_has_arm_neon_f8dot4 and cpuinfo_has_arm_neon_f8mm.

ARM type support refers to scalar ARM, not NEON. FP8 scalar conversions are checked via cpuinfo_has_arm_fp8, while the dot product and matrix multiplication instructions are NEON/SIMD.

Details:

  • Add backward compatibility macros:
    #define cpuinfo_has_arm_f8dot cpuinfo_has_arm_neon_f8dot4
    #define cpuinfo_has_arm_f8mm cpuinfo_has_arm_neon_f8mm
  • Rename inline helpers to cpuinfo_has_arm_neon_f8dot4 and cpuinfo_has_arm_neon_f8mm
  • Update callers in tools/isa-info.c
  • Add test coverage in test/mock/pixel-8.cc

Fixes #427

@meta-cla meta-cla Bot added the cla signed label Jul 23, 2026
Rename cpuinfo_has_arm_f8dot and cpuinfo_has_arm_f8mm to
cpuinfo_has_arm_neon_f8dot4 and cpuinfo_has_arm_neon_f8mm.

ARM type support refers to scalar ARM, not NEON. FP8 scalar conversions
are checked via cpuinfo_has_arm_fp8, while the dot product and matrix
multiplication instructions are NEON/SIMD.

Details:
- Add backward compatibility macros:
  #define cpuinfo_has_arm_f8dot cpuinfo_has_arm_neon_f8dot4
  #define cpuinfo_has_arm_f8mm cpuinfo_has_arm_neon_f8mm
- Rename inline helpers to cpuinfo_has_arm_neon_f8dot4 and cpuinfo_has_arm_neon_f8mm
- Update callers in tools/isa-info.c
- Add test coverage in test/mock/pixel-8.cc

Fixes pytorch#427
@JakeStevens
JakeStevens merged commit 46fbc42 into pytorch:main Jul 23, 2026
12 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

cpuinfo_has_arm_f8dot naming is wrong

2 participants