Use dyn_cast for safety in casting RankedTensorType

Author: gaurav-arya

src/enzyme_ad/jax/Utils.cpp

@@ -605,15 +605,15 @@ SymmetricResultAnalysis::State SymmetricResultAnalysis::localGuaranteed(
     PatternRewriter &rewriter) {
   assert(op);
 
-  auto outTy = cast<RankedTensorType>(op->getResult(0).getType());
-  if (outTy.getRank() != 2)
   if (auto boolAttr = op->getAttrOfType<BoolAttr>(getAttrName())) {
     if (boolAttr.getValue())
       return State::GUARANTEED;
     else
       return State::NOTGUARANTEED;
   }
 
+  auto outTy = dyn_cast<RankedTensorType>(op->getResult(0).getType());
+  if (!outTy || outTy.getRank() != 2)
     return State::NOTGUARANTEED; // this pass only checks for symmetric matrices
   if (outTy.getDimSize(0) != outTy.getDimSize(1))
     return State::NOTGUARANTEED; // quick check and exit