HardwareCapabilities.cs 2.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. using System;
  2. using System.Runtime.Intrinsics.X86;
  3. namespace ARMeilleure.CodeGen.X86
  4. {
  5. static class HardwareCapabilities
  6. {
  7. static HardwareCapabilities()
  8. {
  9. if (!X86Base.IsSupported)
  10. {
  11. return;
  12. }
  13. (int maxNum, _, _, _) = X86Base.CpuId(0x00000000, 0x00000000);
  14. (_, _, int ecx1, int edx1) = X86Base.CpuId(0x00000001, 0x00000000);
  15. FeatureInfo1Edx = (FeatureFlags1Edx)edx1;
  16. FeatureInfo1Ecx = (FeatureFlags1Ecx)ecx1;
  17. if (maxNum >= 7)
  18. {
  19. (_, int ebx7, _, _) = X86Base.CpuId(0x00000007, 0x00000000);
  20. FeatureInfo7Ebx = (FeatureFlags7Ebx)ebx7;
  21. }
  22. }
  23. [Flags]
  24. public enum FeatureFlags1Edx
  25. {
  26. Sse = 1 << 25,
  27. Sse2 = 1 << 26
  28. }
  29. [Flags]
  30. public enum FeatureFlags1Ecx
  31. {
  32. Sse3 = 1 << 0,
  33. Pclmulqdq = 1 << 1,
  34. Ssse3 = 1 << 9,
  35. Fma = 1 << 12,
  36. Sse41 = 1 << 19,
  37. Sse42 = 1 << 20,
  38. Popcnt = 1 << 23,
  39. Aes = 1 << 25,
  40. Avx = 1 << 28,
  41. F16c = 1 << 29
  42. }
  43. [Flags]
  44. public enum FeatureFlags7Ebx
  45. {
  46. Avx2 = 1 << 5,
  47. Sha = 1 << 29
  48. }
  49. public static FeatureFlags1Edx FeatureInfo1Edx { get; }
  50. public static FeatureFlags1Ecx FeatureInfo1Ecx { get; }
  51. public static FeatureFlags7Ebx FeatureInfo7Ebx { get; } = 0;
  52. public static bool SupportsSse => FeatureInfo1Edx.HasFlag(FeatureFlags1Edx.Sse);
  53. public static bool SupportsSse2 => FeatureInfo1Edx.HasFlag(FeatureFlags1Edx.Sse2);
  54. public static bool SupportsSse3 => FeatureInfo1Ecx.HasFlag(FeatureFlags1Ecx.Sse3);
  55. public static bool SupportsPclmulqdq => FeatureInfo1Ecx.HasFlag(FeatureFlags1Ecx.Pclmulqdq);
  56. public static bool SupportsSsse3 => FeatureInfo1Ecx.HasFlag(FeatureFlags1Ecx.Ssse3);
  57. public static bool SupportsFma => FeatureInfo1Ecx.HasFlag(FeatureFlags1Ecx.Fma);
  58. public static bool SupportsSse41 => FeatureInfo1Ecx.HasFlag(FeatureFlags1Ecx.Sse41);
  59. public static bool SupportsSse42 => FeatureInfo1Ecx.HasFlag(FeatureFlags1Ecx.Sse42);
  60. public static bool SupportsPopcnt => FeatureInfo1Ecx.HasFlag(FeatureFlags1Ecx.Popcnt);
  61. public static bool SupportsAesni => FeatureInfo1Ecx.HasFlag(FeatureFlags1Ecx.Aes);
  62. public static bool SupportsAvx => FeatureInfo1Ecx.HasFlag(FeatureFlags1Ecx.Avx);
  63. public static bool SupportsAvx2 => FeatureInfo7Ebx.HasFlag(FeatureFlags7Ebx.Avx2) && SupportsAvx;
  64. public static bool SupportsF16c => FeatureInfo1Ecx.HasFlag(FeatureFlags1Ecx.F16c);
  65. public static bool SupportsSha => FeatureInfo7Ebx.HasFlag(FeatureFlags7Ebx.Sha);
  66. public static bool ForceLegacySse { get; set; }
  67. public static bool SupportsVexEncoding => SupportsAvx && !ForceLegacySse;
  68. }
  69. }