Add asserts

IgWod-IMG · IgWod-IMG · commit ef0629545f95 · 2025-06-09T15:27:00.000+01:00
diff --git a/mlir/lib/Dialect/SPIRV/IR/SPIRVTypes.cpp b/mlir/lib/Dialect/SPIRV/IR/SPIRVTypes.cpp
@@ -245,10 +245,12 @@ Type CooperativeMatrixType::getElementType() const {
 }
 
 uint32_t CooperativeMatrixType::getRows() const {
+  assert(getImpl()->shape[0] != ShapedType::kDynamic);
   return static_cast<uint32_t>(getImpl()->shape[0]);
 }
 
 uint32_t CooperativeMatrixType::getColumns() const {
+  assert(getImpl()->shape[1] != ShapedType::kDynamic);
   return static_cast<uint32_t>(getImpl()->shape[1]);
 }
 

Original file line number	Diff line number	Diff line change
`@@ -245,10 +245,12 @@ Type CooperativeMatrixType::getElementType() const {`
`245`	`245`	`}`
`246`	`246`
`247`	`247`	`uint32_t CooperativeMatrixType::getRows() const {`
	`248`	`+ assert(getImpl()->shape[0] != ShapedType::kDynamic);`
`248`	`249`	`return static_cast<uint32_t>(getImpl()->shape[0]);`
`249`	`250`	`}`
`250`	`251`
`251`	`252`	`uint32_t CooperativeMatrixType::getColumns() const {`
	`253`	`+ assert(getImpl()->shape[1] != ShapedType::kDynamic);`
`252`	`254`	`return static_cast<uint32_t>(getImpl()->shape[1]);`
`253`	`255`	`}`
`254`	`256`