Fix Perf Analyzer CPU-only build (#178)

triton-inference-server · Oct 20, 2022 · aa8f2a2 · aa8f2a2
1 parent 6211f98
commit aa8f2a2
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/src/c++/perf_analyzer/client_backend/triton_c_api/triton_c_api_backend.cc b/src/c++/perf_analyzer/client_backend/triton_c_api/triton_c_api_backend.cc
@@ -153,13 +153,15 @@ TritonCApiClientBackend::RegisterSystemMemory(
   return Error::Success;
 }
 
+#ifdef TRITON_ENABLE_GPU
 Error
 TritonCApiClientBackend::RegisterCudaMemory(
     const std::string& name, void* handle, const size_t byte_size)
 {
   RETURN_IF_ERROR(triton_loader_->RegisterCudaMemory(name, handle, byte_size));
   return Error::Success;
 }
+#endif  // TRITON_ENABLE_GPU
 
 void
 TritonCApiClientBackend::ParseInferInputToTriton(

diff --git a/src/c++/perf_analyzer/client_backend/triton_c_api/triton_c_api_backend.h b/src/c++/perf_analyzer/client_backend/triton_c_api/triton_c_api_backend.h
@@ -106,9 +106,11 @@ class TritonCApiClientBackend : public ClientBackend {
       const std::string& model_name = "",
       const std::string& model_version = "") override;
 
+#ifdef TRITON_ENABLE_GPU
   /// See ClientBackend::RegisterCudaMemory
   Error RegisterCudaMemory(
       const std::string& name, void* handle, const size_t byte_size) override;
+#endif  // TRITON_ENABLE_GPU
 
   /// See ClientBackend::RegisterSystemMemory
   Error RegisterSystemMemory(

diff --git a/src/c++/perf_analyzer/client_backend/triton_c_api/triton_loader.cc b/src/c++/perf_analyzer/client_backend/triton_c_api/triton_loader.cc
@@ -1225,6 +1225,7 @@ TritonLoader::~TritonLoader()
   ClearHandles();
 }
 
+#ifdef TRITON_ENABLE_GPU
 Error
 TritonLoader::RegisterCudaMemory(
     const std::string& name, void* handle, const size_t byte_size)
@@ -1233,6 +1234,7 @@ TritonLoader::RegisterCudaMemory(
       name, handle, byte_size, 0 /* device id */));
   return Error::Success;
 }
+#endif  // TRITON_ENABLE_GPU
 
 Error
 TritonLoader::RegisterSystemMemory(

diff --git a/src/c++/perf_analyzer/client_backend/triton_c_api/triton_loader.h b/src/c++/perf_analyzer/client_backend/triton_c_api/triton_loader.h
@@ -124,8 +124,10 @@ class TritonLoader : public tc::InferenceServerClient {
     return Error::Success;
   }
 
+#ifdef TRITON_ENABLE_GPU
   Error RegisterCudaMemory(
       const std::string& name, void* handle, const size_t byte_size);
+#endif  // TRITON_ENABLE_GPU
 
   Error RegisterSystemMemory(
       const std::string& name, void* ptr, const size_t byte_size);