Fix variable name

triton-inference-server · Jun 7, 2023 · 550cf62 · 550cf62
1 parent fe0f318
commit 550cf62
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/libtorch.cc b/src/libtorch.cc
@@ -1884,7 +1884,7 @@ ModelInstanceState::SetInputTensors(
       const auto torch_dtype =
           ConvertDataTypeToTorchType(batch_input.DataType());
       torch::TensorOptions options{torch_dtype.second};
-      auto updated_options = (memory_type == TRITONSERVER_MEMORY_GPU)
+      auto updated_options = (dst_memory_type == TRITONSERVER_MEMORY_GPU)
                                ? options.device(torch::kCUDA, device_.index())
                                : options.device(torch::kCPU);