fix: add checks for running shardy without raising (#1823)

* fix: add checks for running shardy without raising

* fix: only run in the all case

Author: avik-pal

src/Compiler.jl

@@ -1738,6 +1738,8 @@ function compile_mlir!(
             ()
         end
 
+    legal_to_run_shardy_passes = compile_options.optimization_passes === :all
+
     if compile_options.optimization_passes === :all
         run_pass_pipeline!(
             mod,
@@ -2173,7 +2175,7 @@ function compile_mlir!(
     # shardy passes
     use_shardy_partitioner = false
     result_shardings = missing
-    if is_sharded
+    if is_sharded && legal_to_run_shardy_passes
         module_op = copy(MLIR.IR.Operation(mod))
         mod_copied = MLIR.IR.Module(module_op)