[llvm] 2f48178 - [VirtRegMap] Use Register for Virt2ShapeMap key. NFC

[Craig Topper via llvm-commits]

Author: Craig Topper
Date: 2024-09-15T09:59:34-07:00
New Revision: 2f48178825686283adc6903aae3626cff8d9797c

URL: https://github.com/llvm/llvm-project/commit/2f48178825686283adc6903aae3626cff8d9797c
DIFF: https://github.com/llvm/llvm-project/commit/2f48178825686283adc6903aae3626cff8d9797c.diff

LOG: [VirtRegMap] Use Register for Virt2ShapeMap key. NFC

Added: 
    

Modified: 
    llvm/include/llvm/CodeGen/VirtRegMap.h

Removed: 
    


################################################################################
diff  --git a/llvm/include/llvm/CodeGen/VirtRegMap.h b/llvm/include/llvm/CodeGen/VirtRegMap.h
index 864eb23e133ebd..c9998ba0b47a60 100644
--- a/llvm/include/llvm/CodeGen/VirtRegMap.h
+++ b/llvm/include/llvm/CodeGen/VirtRegMap.h
@@ -63,7 +63,7 @@ class TargetInstrInfo;
 
     /// Virt2ShapeMap - For X86 AMX register whose register is bound shape
     /// information.
-    DenseMap<unsigned, ShapeT> Virt2ShapeMap;
+    DenseMap<Register, ShapeT> Virt2ShapeMap;
 
     /// createSpillSlot - Allocate a spill slot for RC from MFI.
     unsigned createSpillSlot(const TargetRegisterClass *RC);
@@ -123,7 +123,7 @@ class TargetInstrInfo;
     }
 
     void assignVirt2Shape(Register virtReg, ShapeT shape) {
-      Virt2ShapeMap[virtReg.id()] = shape;
+      Virt2ShapeMap[virtReg] = shape;
     }
 
     /// clears the specified virtual register's, physical
@@ -153,7 +153,7 @@ class TargetInstrInfo;
     void setIsSplitFromReg(Register virtReg, Register SReg) {
       Virt2SplitMap[virtReg.id()] = SReg;
       if (hasShape(SReg)) {
-        Virt2ShapeMap[virtReg.id()] = getShape(SReg);
+        Virt2ShapeMap[virtReg] = getShape(SReg);
       }
     }