Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/pytorch/cpuinfo.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'src/arm/mach/init.c')
-rw-r--r--src/arm/mach/init.c8
1 files changed, 5 insertions, 3 deletions
diff --git a/src/arm/mach/init.c b/src/arm/mach/init.c
index d820744..dbea578 100644
--- a/src/arm/mach/init.c
+++ b/src/arm/mach/init.c
@@ -24,7 +24,6 @@
#ifndef CPUFAMILY_ARM_LIGHTNING_THUNDER
#define CPUFAMILY_ARM_LIGHTNING_THUNDER 0x462504D2
#endif
-
#ifndef CPUFAMILY_ARM_FIRESTORM_ICESTORM
#define CPUFAMILY_ARM_FIRESTORM_ICESTORM 0x1B588BB3
#endif
@@ -349,6 +348,7 @@ void cpuinfo_arm_mach_init(void) {
case CPUFAMILY_ARM_MONSOON_MISTRAL:
case CPUFAMILY_ARM_VORTEX_TEMPEST:
case CPUFAMILY_ARM_LIGHTNING_THUNDER:
+ case CPUFAMILY_ARM_FIRESTORM_ICESTORM:
#if CPUINFO_ARCH_ARM64
cpuinfo_isa.atomics = true;
#endif
@@ -360,8 +360,10 @@ void cpuinfo_arm_mach_init(void) {
* ARMv8.2 optional dot-product instructions, so we currently whitelist CPUs
* known to support these instruction.
*/
- if (cpu_family == CPUFAMILY_ARM_LIGHTNING_THUNDER) {
- cpuinfo_isa.dot = true;
+ switch (cpu_family) {
+ case CPUFAMILY_ARM_LIGHTNING_THUNDER:
+ case CPUFAMILY_ARM_FIRESTORM_ICESTORM:
+ cpuinfo_isa.dot = true;
}
uint32_t num_clusters = 1;