--- util/osu_util_mpi.c 2023-01-11 16:10:38.912741000 -0800
+++ util/osu_util_mpi_new.c     2023-01-11 16:10:28.733747000 -0800
@@ -1242,6 +1242,13 @@
             CUDA_CHECK(cudaDeviceSynchronize());
 #endif
             break;
+        case ROCM:
+#ifdef _ENABLE_ROCM_
+            ROCM_CHECK(hipMemcpy((void *)buffer, (void *)temp_buffer,
+                       size * num_procs, hipMemcpyHostToDevice));
+            ROCM_CHECK(hipDeviceSynchronize());
+#endif
+            break;
         default:
             break;
     }
@@ -1485,6 +1492,13 @@
             CUDA_CHECK(cudaDeviceSynchronize());
             break;
 #endif
+#ifdef _ENABLE_ROCM_
+        case ROCM:
+            ROCM_CHECK(hipMemcpy((void *)temp_buffer, (void *)buffer, size *
+                        value2, hipMemcpyDeviceToHost));
+            ROCM_CHECK(hipDeviceSynchronize());
+            break;
+#endif
         default:
             break;
     }