diff --git a/backends/vulkan/runtime/graph/ComputeGraph.cpp b/backends/vulkan/runtime/graph/ComputeGraph.cpp index 5e9c7b7ad2a..b14d0f6ab0b 100644 --- a/backends/vulkan/runtime/graph/ComputeGraph.cpp +++ b/backends/vulkan/runtime/graph/ComputeGraph.cpp @@ -1134,8 +1134,9 @@ void ComputeGraph::clear_deferred_cmds() { void ComputeGraph::prepack() { int i = 0; bool submitted = false; - const bool reduce_peak_memory = total_constant_nbytes_ > 500 * MB; + const bool reduce_peak_memory = total_constant_nbytes_ > 10 * MB; // int count = 0; + context_->set_cmd(); for (std::unique_ptr& node : prepack_nodes_) { // Do not trigger on the first or last prepack node.