Update the implementation for GetUpperBoundGRF

Use #define macros to store the upperBoundGRF entries
Support the tables in MaxGRFTables.* files

Author: iwwu

IGC/Compiler/CMakeLists.txt

@@ -37,6 +37,7 @@ set(IGC_BUILD__SRC__Compiler
     "${CMAKE_CURRENT_SOURCE_DIR}/CodeGenContext.cpp"
     "${CMAKE_CURRENT_SOURCE_DIR}/CodeGenContextWrapper.cpp"
     "${CMAKE_CURRENT_SOURCE_DIR}/compiler_caps.cpp"
+    "${CMAKE_CURRENT_SOURCE_DIR}/MaxGRFTables.cpp"
     "${CMAKE_CURRENT_SOURCE_DIR}/CustomSafeOptPass.cpp"
     "${CMAKE_CURRENT_SOURCE_DIR}/CustomUnsafeOptPass.cpp"
     "${CMAKE_CURRENT_SOURCE_DIR}/CustomLoopOpt.cpp"
@@ -141,6 +142,7 @@ set(IGC_BUILD__HDR__Compiler
     "${CMAKE_CURRENT_SOURCE_DIR}/CodeGenPublic.h"
     "${CMAKE_CURRENT_SOURCE_DIR}/CodeGenPublicEnums.h"
     "${CMAKE_CURRENT_SOURCE_DIR}/compiler_caps.h"
+    "${CMAKE_CURRENT_SOURCE_DIR}/MaxGRFTables.h"
     "${CMAKE_CURRENT_SOURCE_DIR}/CustomSafeOptPass.hpp"
     "${CMAKE_CURRENT_SOURCE_DIR}/CustomUnsafeOptPass.hpp"
     "${CMAKE_CURRENT_SOURCE_DIR}/CustomLoopOpt.hpp"

IGC/Compiler/MaxGRFTables.cpp

@@ -0,0 +1,65 @@
+/*========================== begin_copyright_notice ============================
+
+Copyright (C) 2025 Intel Corporation
+
+SPDX-License-Identifier: MIT
+
+============================= end_copyright_notice ===========================*/
+
+#include "MaxGRFTables.h"
+#include "Compiler/CISACodeGen/Platform.hpp"
+
+namespace IGC
+{
+    bool MaxGRFTable::MatchHWLocalID(HWLocalId fromTable, HWLocalId target)
+    {
+        // If target = '0', table entries with '0 or 1' and '0' are matched
+        // If target = '1', table entries with '0 or 1' and '1' are matched
+        if (fromTable == HWLocalId::EITHER ||
+            fromTable == target)
+            return true;
+        else
+            return false;
+    }
+
+    uint16_t MaxGRFTable::LookupMaxGRF(
+        SIMDMode simt, HWLocalId hwlid, unsigned int witems,
+        const vector<MaxGRFEntry>& table)
+    {
+        MaxGRFEntry entry;
+        // Reverse order intentionally
+        // GRF column is accessed in descening order
+        for (auto it = table.rbegin(); it != table.rend(); ++it)
+        {
+            entry = (*it);
+            if (simt != entry.SIMT)
+                continue;
+
+            if (MatchHWLocalID(entry.HWLID, hwlid) && witems <= entry.Workitems)
+            {
+                return entry.GRF;
+            }
+        }
+
+        return 0;
+    }
+
+    void MaxGRFTable::LoadTable(const CPlatform& platform)
+    {
+        {
+#define GRF_SIMT_LID_WITEMS(a, b, c, d) table.push_back({a, b, c, d});
+            XE3_GRF
+#undef GRF_SIMT_LID_WITEMS
+        }
+    }
+
+    uint16_t MaxGRFTable::GetMaxGRF(SIMDMode simt, HWLocalId hwlid,
+        uint16_t witems, const CPlatform& platform)
+    {
+        if (table.empty())
+            LoadTable(platform);
+
+        return LookupMaxGRF(simt, hwlid, witems, table);
+    }
+
+} //namespace IGC

IGC/Compiler/MaxGRFTables.h

@@ -0,0 +1,78 @@
+/*========================== begin_copyright_notice ============================
+
+Copyright (C) 2025 Intel Corporation
+
+SPDX-License-Identifier: MIT
+
+============================= end_copyright_notice ===========================*/
+
+#pragma once
+
+#include "../common/Types.hpp"
+#include <stdint.h>
+#include <vector>
+
+using namespace std;
+using namespace IGC;
+
+namespace IGC
+{
+    class CPlatform;
+
+    enum class HWLocalId : unsigned char
+    {
+        VALUE0 = 0,
+        VALUE1,
+        EITHER
+    };
+
+    // Define a struct to represent each row
+    struct MaxGRFEntry
+    {
+        uint16_t GRF;
+        SIMDMode SIMT;
+        HWLocalId HWLID; // HW loal-id generation
+        uint16_t Workitems;
+    };
+
+#define XE3_GRF \
+    GRF_SIMT_LID_WITEMS(128, SIMDMode::SIMD16, HWLocalId::EITHER, 1024) \
+    GRF_SIMT_LID_WITEMS(128, SIMDMode::SIMD32, HWLocalId::VALUE1, 1024) \
+    GRF_SIMT_LID_WITEMS(128, SIMDMode::SIMD32, HWLocalId::VALUE0, 2048) \
+    GRF_SIMT_LID_WITEMS(160, SIMDMode::SIMD16, HWLocalId::EITHER,  768) \
+    GRF_SIMT_LID_WITEMS(160, SIMDMode::SIMD32, HWLocalId::VALUE1, 1024) \
+    GRF_SIMT_LID_WITEMS(160, SIMDMode::SIMD32, HWLocalId::VALUE0, 1536) \
+    GRF_SIMT_LID_WITEMS(192, SIMDMode::SIMD16, HWLocalId::EITHER,  640) \
+    GRF_SIMT_LID_WITEMS(192, SIMDMode::SIMD32, HWLocalId::VALUE1, 1024) \
+    GRF_SIMT_LID_WITEMS(192, SIMDMode::SIMD32, HWLocalId::VALUE0, 1280) \
+    GRF_SIMT_LID_WITEMS(256, SIMDMode::SIMD16, HWLocalId::EITHER,  512) \
+    GRF_SIMT_LID_WITEMS(256, SIMDMode::SIMD32, HWLocalId::EITHER, 1024)
+
+
+
+    // Singleton class to create only one instance
+    class MaxGRFTable
+    {
+    private:
+        MaxGRFTable() { }
+        MaxGRFTable(const MaxGRFTable&) = delete;
+        MaxGRFTable& operator=(const MaxGRFTable&) = delete;
+        vector<MaxGRFEntry> table;
+        void LoadTable(const CPlatform& platform);
+        bool MatchHWLocalID(HWLocalId fromTable, HWLocalId target);
+        uint16_t LookupMaxGRF(
+            SIMDMode simt, HWLocalId hwlid, unsigned int witems,
+            const vector<MaxGRFEntry>& table);
+
+    public:
+        // Static method to access the singleton instance
+        static MaxGRFTable& GetInstance()
+        {
+            static MaxGRFTable grfTable;
+            return grfTable;
+        }
+
+        uint16_t GetMaxGRF(SIMDMode simt, HWLocalId hwlid,
+            uint16_t witems, const CPlatform& platform);
+    };
+}