fix: check for ifrt_array_copy_to_host_buffer

Author: avik-pal

src/xla/IFRT/Array.jl

@@ -138,7 +138,8 @@ function XLA.buffer_on_cpu(::Array)
 end
 
 function XLA.to_host(buffer::Array, data, reactant_sharding)
-    if length(XLA.devices(XLA.sharding(buffer))) == 1
+    if is_single_device_sharding(XLA.sharding(buffer)) ||
+        is_fully_replicated(XLA.sharding(buffer))
         GC.@preserve buffer data begin
             @ccall MLIR.API.mlir_c.ifrt_array_copy_to_host_buffer(
                 buffer.buffer::Ptr{Cvoid}, data::Ptr{Cvoid}