[MLIR][AMDGPU] Address TOSA related review comments.

Signed-off-by: Mirza Halilcevic <[email protected]>

Author: mirza-halilcevic

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

@@ -696,8 +696,8 @@ LogicalResult TosaValidation::applyErrorIfCheck(Operation *op) {
 
 bool TosaValidation::isValidElementType(Type type) {
   if (isa<FloatType>(type)) {
-    return isa<Float32Type, Float16Type, BFloat16Type, Float8E4M3FNUZType,
-               Float8E5M2FNUZType, Float8E4M3FNType, Float8E5M2Type>(type);
+    return isa<Float32Type, Float16Type, BFloat16Type, Float8E4M3FNType,
+               Float8E5M2Type>(type);
   } else if (auto intTy = dyn_cast<IntegerType>(type)) {
     if (intTy.isSignless()) {
       switch (intTy.getWidth()) {

mlir/lib/IR/Types.cpp

@@ -44,7 +44,7 @@ bool Type::isF128() const { return llvm::isa<Float128Type>(*this); }
 
 bool Type::isFloat() const { return llvm::isa<FloatType>(*this); }
 
-/// Return true if this is an integer type with the specified width.
+/// Return true if this is a float type with the specified width.
 bool Type::isFloat(unsigned width) const {
   if (auto fltTy = llvm::dyn_cast<FloatType>(*this))
     return fltTy.getWidth() == width;