AMX extension

CMakeLists.txt

@@ -39,7 +39,7 @@ git_submodule_update()
 
 if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "MSVC")
 else()
-SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall -Wextra -O2 -fdata-sections -ffunction-sections")
+SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -mamx-int8 -mamx-tile -Wall -Wextra -O2 -fdata-sections -ffunction-sections")
 endif()
 
 if(CMAKE_SYSTEM_NAME STREQUAL "Darwin")

include/firestarter/Environment/X86/Payload/AVX512_AMX_Payload.hpp

@@ -0,0 +1,58 @@
+/******************************************************************************
+ * FIRESTARTER - A Processor Stress Test Utility
+ * Copyright (C) 2020 TU Dresden, Center for Information Services and High
+ * Performance Computing
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/\>.
+ *
+ * Contact: daniel.hackenberg@tu-dresden.de
+ *****************************************************************************/
+
+#pragma once
+
+#include <firestarter/Environment/X86/Payload/X86Payload.hpp>
+
+namespace firestarter::environment::x86::payload {
+class AVX512_AMX_Payload final : public X86Payload {
+public:
+  AVX512_AMX_Payload(asmjit::CpuFeatures const &supportedFeatures)
+      : X86Payload(supportedFeatures, {asmjit::CpuFeatures::X86::kAMX_BF16},
+                   "AVX512_AMX", 8, 32) {}
+
+  int compilePayload(
+      std::vector<std::pair<std::string, unsigned>> const &proportion,
+      unsigned instructionCacheSize,
+      std::list<unsigned> const &dataCacheBufferSize, unsigned ramBufferSize,
+      unsigned thread, unsigned numberOfLines, bool dumpRegisters,
+      bool errorDetection) override;
+  std::list<std::string> getAvailableInstructions() const override;
+  void init(unsigned long long *memoryAddr,
+            unsigned long long bufferSize) override;
+
+  firestarter::environment::payload::Payload *clone() const override {
+    return new AVX512_AMX_Payload(this->supportedFeatures());
+  };
+
+private:
+  const std::map<std::string, unsigned> instructionFlops = {
+      {"REG", 32},   {"L1_L", 32},  {"L1_BROADCAST", 16}, {"L1_S", 16},
+      {"L1_LS", 16}, {"L2_L", 32},  {"L2_S", 16},         {"L2_LS", 16},
+      {"L3_L", 32},  {"L3_S", 16},  {"L3_LS", 16},        {"L3_P", 16},
+      {"RAM_L", 32}, {"RAM_S", 16}, {"RAM_LS", 16},       {"RAM_P", 16},
+	  {"AMX", 512}};
+
+  const std::map<std::string, unsigned> instructionMemory = {
+      {"RAM_L", 64}, {"RAM_S", 128}, {"RAM_LS", 128}, {"RAM_P", 64}};
+};
+} // namespace firestarter::environment::x86::payload

include/firestarter/Environment/X86/Platform/SapphireRapidsConfig.hpp

@@ -0,0 +1,52 @@
+/******************************************************************************
+ * FIRESTARTER - A Processor Stress Test Utility
+ * Copyright (C) 2020 TU Dresden, Center for Information Services and High
+ * Performance Computing
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/\>.
+ *
+ * Contact: daniel.hackenberg@tu-dresden.de
+ *****************************************************************************/
+
+#pragma once
+
+#include <firestarter/Environment/X86/Payload/AVX512_AMX_Payload.hpp>
+#include <firestarter/Environment/X86/Platform/X86PlatformConfig.hpp>
+
+namespace firestarter::environment::x86::platform {
+class SapphireRapidsConfig final : public X86PlatformConfig {
+
+public:
+  SapphireRapidsConfig(asmjit::CpuFeatures const &supportedFeatures,
+                  unsigned family, unsigned model, unsigned threads)
+      : X86PlatformConfig("SKL_XEONEP", 6, {85}, {1, 2}, 0,
+                          {32768, 1048576, 1441792}, 1048576000, 1536, family,
+                          model, threads,
+                          new payload::AVX512_AMX_Payload(supportedFeatures)) {}
+
+  std::vector<std::pair<std::string, unsigned>>
+  getDefaultPayloadSettings() const override {
+    return std::vector<std::pair<std::string, unsigned>>({{"RAM_S", 3},
+                                                          {"RAM_P", 1},
+                                                          {"L3_S", 1},
+                                                          {"L3_P", 1},
+                                                          {"L2_S", 4},
+                                                          {"L2_L", 70},
+                                                          {"L1_S", 0},
+                                                          {"L1_L", 40},
+                                                          {"REG", 140},
+							  {"AMX", 1}});
+  }
+};
+} // namespace firestarter::environment::x86::platform

include/firestarter/Environment/X86/X86Environment.hpp

@@ -24,6 +24,7 @@
 #include <firestarter/Environment/Environment.hpp>
 #include <firestarter/Environment/X86/X86CPUTopology.hpp>
 
+#include <firestarter/Environment/X86/Platform/SapphireRapidsConfig.hpp>
 #include <firestarter/Environment/X86/Platform/BulldozerConfig.hpp>
 #include <firestarter/Environment/X86/Platform/HaswellConfig.hpp>
 #include <firestarter/Environment/X86/Platform/HaswellEPConfig.hpp>
@@ -88,14 +89,16 @@ class X86Environment final : public Environment {
           REGISTER(HaswellEPConfig),      REGISTER(SandyBridgeConfig),
           REGISTER(SandyBridgeEPConfig),  REGISTER(NehalemConfig),
           REGISTER(NehalemEPConfig),      REGISTER(BulldozerConfig),
-          REGISTER(NaplesConfig),         REGISTER(RomeConfig)};
+          REGISTER(NaplesConfig),         REGISTER(RomeConfig),
+          REGISTER(SapphireRapidsConfig)};
 
   std::list<platform::X86PlatformConfig *> platformConfigs;
 
   // List of fallback PlatformConfig. Add one for each x86 extension.
   const std::list<std::function<platform::X86PlatformConfig *(
       asmjit::CpuFeatures const &, unsigned, unsigned, unsigned)>>
       fallbackPlatformConfigsCtor = {
+	  REGISTER(SapphireRapidsConfig), // AMX + AVX512
           REGISTER(SkylakeSPConfig),   // AVX512
           REGISTER(BulldozerConfig),   // FMA4
           REGISTER(HaswellConfig),     // FMA

src/CMakeLists.txt

@@ -15,6 +15,7 @@ SET(FIRESTARTER_FILES
 
 	firestarter/Environment/X86/Payload/X86Payload.cpp
 	firestarter/Environment/X86/Payload/AVX512Payload.cpp
+	firestarter/Environment/X86/Payload/AVX512_AMX_Payload.cpp
 	firestarter/Environment/X86/Payload/FMA4Payload.cpp
 	firestarter/Environment/X86/Payload/FMAPayload.cpp
 	firestarter/Environment/X86/Payload/ZENFMAPayload.cpp