From b9e4bd89c36184b236ce46720f77912765e42111 Mon Sep 17 00:00:00 2001 From: Iman Tabrizian Date: Tue, 9 Aug 2022 11:40:00 -0400 Subject: [PATCH] Fix CPU only build --- src/libtorch.cc | 4 ---- 1 file changed, 4 deletions(-) diff --git a/src/libtorch.cc b/src/libtorch.cc index 76567dc..cc3ab55 100644 --- a/src/libtorch.cc +++ b/src/libtorch.cc @@ -554,12 +554,9 @@ class ModelInstanceState : public BackendModelInstance { // If the model supports batching. bool supports_batching_; -#ifdef TRITON_ENABLE_GPU - // PyTorch stream used for execution of inferences. cudaEvent_t compute_input_start_event_; cudaEvent_t compute_infer_start_event_; cudaEvent_t compute_output_start_event_; -#endif }; TRITONSERVER_Error* @@ -616,7 +613,6 @@ ModelInstanceState::ModelInstanceState( } } - // If this is a sequence model then make sure that the required // inputs are present in the model and have the correct shape and // datatype.