Fix stablehlo constant folding bugs through patch (#401)

This PR provides a patch to fix two issues in StableHlo
upstream,
- If some op is constant folded and op users are certain
stablehlo ops (which don't allow different type for operand
and result) OR ops with different dialect, `tensor.cast`
should be inserted.
MLIR-TensorRT constant folding pass does this but upstream
stablehlo pass doesn't do this, causing error. Some
logic from MLIR-TRT is patched into upstream.
- Upstream `ReshapeOpCanon` pattern to simplify
`stablehlo.reshape` op failed if input to reshape
is `DenseResourceElementsAttr`. This is fixed.
- Many patterns where replacing op with a new op,
without giving consideration to casting. This is 
fixed.

Solves: #378

Author: shelkesagar29

mlir-tensorrt/CMakeLists.txt

@@ -183,7 +183,9 @@ if(MLIR_TRT_ENABLE_HLO AND NOT TARGET StablehloOps)
     VERSION 1.6.4
     GIT_TAG 1456dfa1e1a83aab0cc717714ba3695886f60302
     GIT_REPOSITORY "https://github.com/openxla/stablehlo.git"
-    PATCHES "${MLIR_TENSORRT_ROOT_DIR}/build_tools/stablehlo.patch"
+    PATCHES 
+      "${MLIR_TENSORRT_ROOT_DIR}/build_tools/stablehlo.patch"
+      "${MLIR_TENSORRT_ROOT_DIR}/build_tools/stablehlo_aggressive_folder.patch"
     OPTIONS
       "STABLEHLO_ENABLE_BINDINGS_PYTHON ${MLIR_TRT_ENABLE_PYTHON}"
       "STABLEHLO_BUILD_EMBEDDED ON"