[LowerMatrixIntrinsics] Call getRegisterClassForType before getNumberOfRegisters.

topperc · topperc · commit 38b30eb2b218 · 2022-01-10T15:32:13.000-08:00
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
diff --git 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) {

Original file line number	Diff line number	Diff line change
`@@ -1391,7 +1391,8 @@ class LowerMatrixIntrinsics {`
`1391`	`1391`	`// reloads necessary.`
`1392`	`1392`	`unsigned Op0Regs = (R + VF - 1) / VF * M;`
`1393`	`1393`	`unsigned Op1Regs = (M + VF - 1) / VF * C;`
`1394`		`- return Op0Regs + Op1Regs > TTI.getNumberOfRegisters(true);`
	`1394`	`+ return Op0Regs + Op1Regs >`
	`1395`	`+ TTI.getNumberOfRegisters(TTI.getRegisterClassForType(true));`
`1395`	`1396`	`}`
`1396`	`1397`
`1397`	`1398`	`MatrixTy getZeroMatrix(Type *EltType, unsigned R, unsigned C) {`