Skip to content

Commit 79f2bf7

Browse files
jan-wassenbergcopybara-github
authored andcommitted
Disable SVE (except SVE2_128) for MatMul due to compiler crash
PiperOrigin-RevId: 886190686
1 parent 90f3de7 commit 79f2bf7

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

compression/types.h

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,9 @@ namespace gcpp {
4242
#define HWY_DISABLED_TARGETS (HWY_SCALAR | HWY_ALL_NEON)
4343
#elif HWY_ARCH_ARM_A64
4444
// We do not yet use AES (e.g. for random generation), hence NEON is the same
45-
// as NEON_WITHOUT_AES. Also skip SVE because SVE2_128 and SVE_256 cover most.
46-
#define GEMMA_DISABLED_TARGETS (HWY_SCALAR | HWY_NEON | HWY_SVE)
45+
// as NEON_WITHOUT_AES. Also skip SVE except SVE2_128 due to compiler bugs.
46+
#define GEMMA_DISABLED_TARGETS \
47+
(HWY_SCALAR | HWY_NEON | HWY_SVE | HWY_SVE2 | HWY_SVE_256)
4748
#elif HWY_ARCH_X86
4849
// Skip anything older than Haswell (2013); use Zen4/SPR for recent CPUs.
4950
// Although we do not use SPR's F16, Zen4 is only enabled for AMD. We do not

0 commit comments

Comments
 (0)