Skip to content

Commit 491bc02

Browse files
committed
Check for DOTPROD
1 parent 9c4f6c4 commit 491bc02

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

lorann/quant.h

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -96,10 +96,10 @@ LORANN_ALWAYS_INLINE inline __m512i dpbusd(__m512i c, const __m512i a, const __m
9696

9797
#endif
9898

99-
#if (defined(__ARM_NEON) || defined(__ARM_NEON__))
99+
#if (defined(__ARM_NEON) || defined(__ARM_NEON__)) && defined(__ARM_FEATURE_DOTPROD)
100100
#pragma GCC push_options
101-
#pragma GCC target("arch=armv8.2-a+dotprod+i8mm")
102-
#pragma clang attribute push(__attribute__((target("arch=armv8.2-a+dotprod+i8mm"))), \
101+
#pragma GCC target("arch=armv8.2-a+dotprod")
102+
#pragma clang attribute push(__attribute__((target("arch=armv8.2-a+dotprod"))), \
103103
apply_to = function)
104104
#endif
105105

@@ -1471,9 +1471,9 @@ struct SQ8Quantizer : SQQuantizer {
14711471
}
14721472
};
14731473

1474-
#if (defined(__ARM_NEON) || defined(__ARM_NEON__))
1474+
#if (defined(__ARM_NEON) || defined(__ARM_NEON__)) && defined(__ARM_FEATURE_DOTPROD)
14751475
#pragma clang attribute pop
14761476
#pragma GCC pop_options
14771477
#endif
14781478

1479-
} // namespace Lorann
1479+
} // namespace Lorann

0 commit comments

Comments
 (0)