Don't use pinned memory when unnecessary (#17832)
Reports of excessive GPU memory usage and errors from cudaHostRegister. There are some cases where pinning is not required.
This commit is contained in:
@@ -2051,7 +2051,8 @@ impl ClusterInfo {
|
||||
.process_pull_requests(callers.cloned(), timestamp());
|
||||
let output_size_limit =
|
||||
self.update_data_budget(stakes.len()) / PULL_RESPONSE_MIN_SERIALIZED_SIZE;
|
||||
let mut packets = Packets::new_with_recycler(recycler.clone(), 64, "handle_pull_requests");
|
||||
let mut packets =
|
||||
Packets::new_unpinned_with_recycler(recycler.clone(), 64, "handle_pull_requests");
|
||||
let (caller_and_filters, addrs): (Vec<_>, Vec<_>) = {
|
||||
let mut rng = rand::thread_rng();
|
||||
let check_pull_request =
|
||||
@@ -2323,7 +2324,7 @@ impl ClusterInfo {
|
||||
None
|
||||
} else {
|
||||
let packets =
|
||||
Packets::new_with_recycler_data(recycler, "handle_ping_messages", packets);
|
||||
Packets::new_unpinned_with_recycler_data(recycler, "handle_ping_messages", packets);
|
||||
Some(packets)
|
||||
}
|
||||
}
|
||||
|
@@ -54,6 +54,7 @@ impl GossipService {
|
||||
Recycler::default(),
|
||||
"gossip_receiver",
|
||||
1,
|
||||
false,
|
||||
);
|
||||
let (response_sender, response_receiver) = channel();
|
||||
let (consume_sender, listen_receiver) = channel();
|
||||
|
Reference in New Issue
Block a user