Update block size enum value to reflect the spec

Change-Id: I8bc0955f994b0602e719b4c060c389bf3950f133

Author: lhutton1

mlir/include/mlir/Dialect/Tosa/IR/TosaOpBase.td

@@ -464,17 +464,14 @@ def Tosa_RoundingModeAttr
     : Tosa_I32EnumAttr<"RoundingMode", "Supported rounding modes", "rounding_mode",
                     [Tosa_ROUNDING_SINGLE_ROUND, Tosa_ROUNDING_INEXACT_ROUND, Tosa_ROUNDING_DOUBLE_ROUND]>;
 
-def Tosa_BLOCK_SIZE_32 : I32EnumAttrCase<"BLOCK_SIZE_32", 1>;
+def Tosa_BLOCK_SIZE_32 : I32EnumAttrCase<"BLOCK_SIZE_32", 32>;
 
 def Tosa_BlockSizeAttr
     : Tosa_I32EnumAttr<"BlockSize", "Block size for the block_scaled formats", "block_size",
                     [Tosa_BLOCK_SIZE_32]> {
   let extraClassDeclaration = [{
-    static unsigned int getBlockSizeValue(BlockSize blockSize) {
-      switch (blockSize) {
-        case BlockSize::BLOCK_SIZE_32:
-          return 32;
-      }
+    static uint32_t getBlockSizeValue(BlockSize blockSize) {
+      return static_cast<uint32_t>(blockSize);
     }
   }];
 }

mlir/lib/Dialect/Tosa/IR/TosaOps.cpp

@@ -1992,8 +1992,7 @@ LogicalResult MatmulTBlockScaledOp::verify() {
            << D << " vs N=" << N;
 
   // Verify C is a multiple of block size
-  const unsigned int blockSize =
-      BlockSizeAttr::getBlockSizeValue(getBlockSize());
+  const uint32_t blockSize = BlockSizeAttr::getBlockSizeValue(getBlockSize());
   if (ShapedType::isStatic(C) && C % blockSize != 0)
     return emitOpError("expect C to be a multiple of block size, got C=")
            << C << ", block_size=" << blockSize;