[llvm] 38b30eb - [LowerMatrixIntrinsics] Call getRegisterClassForType before getNumberOfRegisters.

[Craig Topper via llvm-commits]

Author: Craig Topper
Date: 2022-01-10T15:32:13-08:00
New Revision: 38b30eb2b218ad481855d5c691ef4501711ca3a6

URL: https://github.com/llvm/llvm-project/commit/38b30eb2b218ad481855d5c691ef4501711ca3a6
DIFF: https://github.com/llvm/llvm-project/commit/38b30eb2b218ad481855d5c691ef4501711ca3a6.diff

LOG: [LowerMatrixIntrinsics] Call getRegisterClassForType before getNumberOfRegisters.

getNumberOfRegisters takes a ClassID as it's argument. It shouldn't be passed a bool. Assuming the bool meant vector or not, we should call getRegisterClassForType first.

Reviewed By: fhahn

Differential Revision: https://reviews.llvm.org/D116903

Added: 
    

Modified: 
    llvm/lib/Transforms/Scalar/LowerMatrixIntrinsics.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Transforms/Scalar/LowerMatrixIntrinsics.cpp b/llvm/lib/Transforms/Scalar/LowerMatrixIntrinsics.cpp
index 5723f1d564360..8f1d0181ee5b1 100644
--- a/llvm/lib/Transforms/Scalar/LowerMatrixIntrinsics.cpp
+++ b/llvm/lib/Transforms/Scalar/LowerMatrixIntrinsics.cpp
@@ -1391,7 +1391,8 @@ class LowerMatrixIntrinsics {
     // reloads necessary.
     unsigned Op0Regs = (R + VF - 1) / VF * M;
     unsigned Op1Regs = (M + VF - 1) / VF * C;
-    return Op0Regs + Op1Regs > TTI.getNumberOfRegisters(true);
+    return Op0Regs + Op1Regs >
+           TTI.getNumberOfRegisters(TTI.getRegisterClassForType(true));
   }
 
   MatrixTy getZeroMatrix(Type *EltType, unsigned R, unsigned C) {