Revert "[TOSA] Make validation pass isValidElementType check more str…

…ict (llvm#119671)"

This reverts commit 9472c5f.

mlir/lib/Dialect/Tosa/Transforms/TosaValidation.cpp

@@ -524,8 +524,18 @@ bool TosaValidation::isValidElementType(Type type) {
     if (!isEnabledProfile(TosaProfileEnum::MainInference))
       return false;
     return type.isF32() || type.isF16() || type.isBF16();
-  } else if (auto intTy = dyn_cast<IntegerType>(type)) {
-    if (intTy.isSignless()) {
+  }
+  if (auto intTy = dyn_cast<IntegerType>(type)) {
+    if (intTy.isUnsigned()) {
+      switch (intTy.getWidth()) {
+      case 8:
+      case 16:
+        return true;
+      default:
+        return false;
+      }
+    } else {
+      // Signless - treated as signed.
       switch (intTy.getWidth()) {
       case 1:
       case 4:
@@ -534,10 +544,13 @@ bool TosaValidation::isValidElementType(Type type) {
       case 32:
       case 48:
         return true;
+      default:
+        return false;
       }
     }
+    return false;
   }
-  return false;
+  return true;
 }
 
 void TosaValidation::runOnOperation() {

mlir/test/Dialect/Tosa/level_check.mlir

@@ -143,14 +143,6 @@ func.func @test_const_f64(%arg0 : tensor<1xf64>) {
 
 // -----
 
-func.func @test_const_ui8(%arg0 : tensor<1xui8>) {
-  // expected-error@+1 {{'tosa.const' op is not profile-aligned: element type 'ui8' is not legal}}
-  %0 = "tosa.const"() {value = dense<0> : tensor<1xui8>} : () -> tensor<1xui8>
-  return
-}
-
-// -----
-
 func.func @test_avgpool2d_kernel_y(%arg0: tensor<1x32x32x8xf32>) -> tensor<1x32x32x8xf32> {
   // expected-error@+1 {{'tosa.avg_pool2d' op failed level check: kernel <= MAX_KERNEL}}
   %0 = "tosa.avg_pool2d"(%arg0) {kernel = array<i64: 8193, 1>, pad = array<i64: 4, 4, 4, 4>, stride = array<i64: 1, 1>, acc_type = f32} :