Adding guards around frontend-specific code

triton-inference-server · Oct 28, 2024 · 46121ff · 46121ff
1 parent d2939cb
commit 46121ff
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/src/python/tritonfrontend/_c/tritonfrontend.h b/src/python/tritonfrontend/_c/tritonfrontend.h
@@ -116,17 +116,28 @@ class TritonFrontend {
 
     server_.reset(server_ptr, EmptyDeleter);
 
-    if constexpr (
-        std::is_same_v<FrontendServer, HTTPAPIServer> ||
-        std::is_same_v<FrontendServer, triton::server::grpc::Server>) {
+#ifdef TRITON_ENABLE_HTTP
+    if constexpr (std::is_same_v<FrontendServer, HTTPAPIServer>) {
       ThrowIfError(FrontendServer::Create(
           server_, data, nullptr /* TraceManager */,
           nullptr /* SharedMemoryManager */, restricted_features, &service));
     }
+#endif
 
+#ifdef TRITON_ENABLE_GRPC
+    if constexpr (std::is_same_v<
+                      FrontendServer, triton::server::grpc::Server>) {
+      ThrowIfError(FrontendServer::Create(
+          server_, data, nullptr /* TraceManager */,
+          nullptr /* SharedMemoryManager */, restricted_features, &service));
+    }
+#endif
+
+#ifdef TRITON_ENABLE_METRICS
     if constexpr (std::is_same_v<FrontendServer, HTTPMetricsServer>) {
       ThrowIfError(FrontendServer::Create(server_, data, &service));
     }
+#endif
   };
 
   // TODO: [DLIS-7194] Add support for TraceManager & SharedMemoryManager