Skip to content

Commit bee8fde

Browse files
authored
Fix CPU only build (#75)
1 parent 8c1c6bd commit bee8fde

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

src/libtorch.cc

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -554,12 +554,9 @@ class ModelInstanceState : public BackendModelInstance {
554554
// If the model supports batching.
555555
bool supports_batching_;
556556

557-
#ifdef TRITON_ENABLE_GPU
558-
// PyTorch stream used for execution of inferences.
559557
cudaEvent_t compute_input_start_event_;
560558
cudaEvent_t compute_infer_start_event_;
561559
cudaEvent_t compute_output_start_event_;
562-
#endif
563560
};
564561

565562
TRITONSERVER_Error*
@@ -616,7 +613,6 @@ ModelInstanceState::ModelInstanceState(
616613
}
617614
}
618615

619-
620616
// If this is a sequence model then make sure that the required
621617
// inputs are present in the model and have the correct shape and
622618
// datatype.

0 commit comments

Comments
 (0)