[mlir][sparsifier] fix isAdmissibleBSR (llvm#72195)

makslevental · web-flow · commit e35b60628093 · 2023-11-14T16:56:34.000-06:00
Fixes llvm#72194.
diff --git a/mlir/lib/Dialect/SparseTensor/Transforms/SparseGPUCodegen.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/SparseGPUCodegen.cpp
@@ -436,7 +436,7 @@ static bool isAdmissibleBSR(SparseTensorType &aTp) {
     // CuSparse only supports "square" blocks currently.
     SmallVector<unsigned> dims = getBlockSize(aTp.getDimToLvl());
     assert(dims.size() == 2);
-    return dims[0] = dims[1] && dims[0] > 1;
+    return dims[0] == dims[1] && dims[0] > 1;
   }
   return false;
 }

Original file line number	Diff line number	Diff line change
`@@ -436,7 +436,7 @@ static bool isAdmissibleBSR(SparseTensorType &aTp) {`
`436`	`436`	`// CuSparse only supports "square" blocks currently.`
`437`	`437`	`SmallVector<unsigned> dims = getBlockSize(aTp.getDimToLvl());`
`438`	`438`	`assert(dims.size() == 2);`
`439`		`- return dims[0] = dims[1] && dims[0] > 1;`
	`439`	`+ return dims[0] == dims[1] && dims[0] > 1;`
`440`	`440`	`}`
`441`	`441`	`return false;`
`442`	`442`	`}`