         d_n_unique,
         n_recv);
     alloc.deallocate((char *) d_temp_storage);
- 
+
     hipMemcpy(&n_unique, d_n_unique, sizeof(unsigned int), hipMemcpyDeviceToHost);
     alloc.deallocate((char *)d_n_unique);
 

        d_n_send,
        n_groups);
    alloc.deallocate((char *) d_temp_storage);
 
    hipMemcpy(&n_send, d_n_send, sizeof(unsigned int), hipMemcpyDeviceToHost);
    alloc.deallocate((char *)d_n_send);
    }

        d_n_out,
        n_groups);
    alloc.deallocate((char *) d_temp_storage);

    hipMemcpy(&n_out, d_n_out, sizeof(unsigned int), hipMemcpyDeviceToHost);
    alloc.deallocate((char *)d_n_out);
    }
