Enable raised comms on post sdy propagation (#1516)

Author: wsmoses

src/Compiler.jl

@@ -1611,7 +1611,7 @@ function compile_mlir!(
 
     recognize_comms = true
     lower_comms = true
-    if is_sharded && compile_options.shardy_passes == :to_mhlo_shardings
+    if is_sharded && (compile_options.shardy_passes == :to_mhlo_shardings || compile_options.shardy_passes == :post_sdy_propagation || compile_options.shardy_passes isa ShardyPropagationOptions)
         lower_comms = false
     end