Skip to content

Commit 550cf62

Browse files
dyastremskymc-nv
authored andcommitted
Fix variable name
1 parent fe0f318 commit 550cf62

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/libtorch.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1884,7 +1884,7 @@ ModelInstanceState::SetInputTensors(
18841884
const auto torch_dtype =
18851885
ConvertDataTypeToTorchType(batch_input.DataType());
18861886
torch::TensorOptions options{torch_dtype.second};
1887-
auto updated_options = (memory_type == TRITONSERVER_MEMORY_GPU)
1887+
auto updated_options = (dst_memory_type == TRITONSERVER_MEMORY_GPU)
18881888
? options.device(torch::kCUDA, device_.index())
18891889
: options.device(torch::kCPU);
18901890

0 commit comments

Comments
 (0)