[NFC] Generalize some DotTrait handling (#4098)

Remove some unnecessary restrictions on code related to DotLike.

Only Hopper Dot can take shared memory operands so remove some redundant
checks.

include/triton/Dialect/Triton/IR/Traits.h

@@ -71,7 +71,7 @@ class DotLike : public TraitBase<ConcreteType, DotLike> {
       return op->emitOpError("expected 3 operands");
     auto aTy = cast<TensorOrMemDesc>(op->getOperand(0).getType());
     auto bTy = cast<TensorOrMemDesc>(op->getOperand(1).getType());
-    auto cTy = cast<TensorType>(op->getOperand(2).getType());
+    auto cTy = cast<TensorOrMemDesc>(op->getOperand(2).getType());
     auto aShape = aTy.getShape();
     auto bShape = bTy.getShape();
     auto cShape = cTy.getShape();

lib/Dialect/TritonGPU/Transforms/OptimizeDotOperands.cpp

@@ -214,11 +214,6 @@ class FuseTransHopper : public OpRewritePattern<LocalAllocOp> {
       return failure();
 
     auto dot = *allocOp->getUsers().begin();
-    auto dotEnc = dyn_cast<NvidiaMmaEncodingAttr>(
-        cast<RankedTensorType>(dot->getResult(0).getType()).getEncoding());
-    if (!dotEnc || dotEnc.getVersionMajor() != 3)
-      return failure();
-
     if (!allocOp.getSrc())
       return failure();