Skip to content

Commit 018fa28

Browse files
authored
[ET-VK] Lower reduce_peak_memory threshold from 500 MB to 10 MB
Differential Revision: D100332227 Pull Request resolved: #18816
1 parent 930ecfd commit 018fa28

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

backends/vulkan/runtime/graph/ComputeGraph.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1134,8 +1134,9 @@ void ComputeGraph::clear_deferred_cmds() {
11341134
void ComputeGraph::prepack() {
11351135
int i = 0;
11361136
bool submitted = false;
1137-
const bool reduce_peak_memory = total_constant_nbytes_ > 500 * MB;
1137+
const bool reduce_peak_memory = total_constant_nbytes_ > 10 * MB;
11381138
// int count = 0;
1139+
11391140
context_->set_cmd();
11401141
for (std::unique_ptr<PrepackNode>& node : prepack_nodes_) {
11411142
// Do not trigger on the first or last prepack node.

0 commit comments

Comments
 (0)