diff --git a/backends/vulkan/runtime/graph/ops/PrepackNode.cpp b/backends/vulkan/runtime/graph/ops/PrepackNode.cpp index 60d1982d97e..192d7496045 100644 --- a/backends/vulkan/runtime/graph/ops/PrepackNode.cpp +++ b/backends/vulkan/runtime/graph/ops/PrepackNode.cpp @@ -36,8 +36,8 @@ void PrepackNode::encode(ComputeGraph* graph) { api::Context* const context = graph->context(); api::PipelineBarrier pipeline_barrier{}; - TensorRef tref = graph->get_val(tref_).toTensorRef(); - vTensor packed = graph->get_val(packed_).toTensor(); + TensorRef& tref = graph->get_val(tref_).toTensorRef(); + vTensor& packed = graph->get_val(packed_).toTensor(); size_t numel = api::utils::multiply_integers(tref.sizes); api::StorageBuffer staging(graph->context(), tref.dtype, numel); diff --git a/backends/vulkan/runtime/graph/ops/impl/Staging.cpp b/backends/vulkan/runtime/graph/ops/impl/Staging.cpp index a506564a018..e08fad5df83 100644 --- a/backends/vulkan/runtime/graph/ops/impl/Staging.cpp +++ b/backends/vulkan/runtime/graph/ops/impl/Staging.cpp @@ -65,7 +65,7 @@ ValueRef prepack( const api::GPUMemoryLayout layout) { TensorRef& tref = graph.get_val(vref).toTensorRef(); ValueRef v = graph.add_tensor(tref.sizes, tref.dtype, layout); - vTensor t = graph.get_val(v).toTensor(); + vTensor& t = graph.get_val(v).toTensor(); api::ShaderInfo shader = get_nchw_to_image_shader(t); diff --git a/backends/vulkan/runtime/graph/ops/impl/Sum.cpp b/backends/vulkan/runtime/graph/ops/impl/Sum.cpp index c4ba07ae4a0..a8cd31c5d61 100644 --- a/backends/vulkan/runtime/graph/ops/impl/Sum.cpp +++ b/backends/vulkan/runtime/graph/ops/impl/Sum.cpp @@ -120,7 +120,7 @@ void add_sum_dim_IntList( vTensor& in_tensor = graph.get_val(in).toTensor(); std::set dims_set; - auto dims_to_sum = graph.get_val(opt_dim).toIntList(); + const auto& dims_to_sum = graph.get_val(opt_dim).toIntList(); int64_t in_dim = in_tensor.sizes().size(); for (const auto& dim : dims_to_sum) { diff --git a/backends/vulkan/test/vulkan_compute_api_test.cpp b/backends/vulkan/test/vulkan_compute_api_test.cpp index 3265191180b..28166502239 100644 --- a/backends/vulkan/test/vulkan_compute_api_test.cpp +++ b/backends/vulkan/test/vulkan_compute_api_test.cpp @@ -401,7 +401,7 @@ TEST(VulkanComputeGraphTest, test_values_scalar_list_inplace_constructed) { ComputeGraph graph(config); ValueRef idx = graph.add_scalar_list({1, 2, 3, 4}); - std::vector& arr = graph.get_val(idx).toIntList(); + const auto& arr = graph.get_val(idx).toIntList(); EXPECT_TRUE(arr.size() == 4); for (int i = 0; i < 4; i++) { EXPECT_TRUE(arr[i] == i + 1); @@ -417,7 +417,7 @@ TEST(VulkanComputeGraphTest, test_values_scalar_list_outside_constructed) { std::vector data = {5.0, 4.0, 3.0, 2.0, 1.0}; idx = graph.add_scalar_list(std::move(data)); } - std::vector& arr = graph.get_val(idx).toDoubleList(); + const auto& arr = graph.get_val(idx).toDoubleList(); EXPECT_TRUE(arr.size() == 5); for (int i = 0; i < 5; i++) { EXPECT_TRUE(arr[i] == (5 - i));