mirrors/Terra
1
0
Fork 0
mirror of https://github.com/PolyhedralDev/Terra.git synced 2026-04-05 23:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

detect fma

Browse Source
This commit is contained in:
Zoë Gidiere
2023-10-07 22:44:19 -06:00
parent f384d09376
commit dcddc0504e
4 changed files with 45 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
common/api/src/main/java/com/dfsek/terra/api/util/MathUtil.java
View File

@@ -7,6 +7,11 @@
package com.dfsek.terra.api.util;
import cpufeatures.CpuFeatures;
import cpufeatures.aarch64.Aarch64Feature;
import cpufeatures.arm.ArmFeature;
import cpufeatures.x86.X86Feature;
import java.util.List;
@@ -19,6 +24,26 @@ public final class MathUtil {
private static final double degFull, degToIndex;
private static final double[] sin, cos;
public static final boolean hasFMA;
static {
boolean configHasFMA = false;
try {
CpuFeatures.load();
configHasFMA = switch (CpuFeatures.getArchitecture()) {
case UNSUPPORTED -> false;
case AARCH64 -> CpuFeatures.getAarch64Info().has(Aarch64Feature.ASIMD);
case ARM -> CpuFeatures.getArmInfo().has(ArmFeature.NEON);
case RISCV -> true;
case X86 -> CpuFeatures.getX86Info().has(X86Feature.FMA3);
};
} catch (Throwable t) {
System.err.println("Failed to determine FMA support, assuming no FMA support");
}
hasFMA = configHasFMA;
System.out.println("FMA: " + configHasFMA);
}
static {
SIN_BITS = 12;
SIN_MASK = ~(-1 << SIN_BITS);
@@ -259,4 +284,12 @@ public final class MathUtil {
public static double hypot(double x, double y) {
return Math.sqrt(Math.pow(x, 2) + Math.pow(y, 2));
}
public static double fma(double a, double b, double c) {
if (hasFMA) {
return Math.fma(a, b, c);
} else {
return a * b + c;
}
}
}