diff --git a/mediapipe/framework/calculator_graph.h b/mediapipe/framework/calculator_graph.h
index 8d58ff312..04f9de45f 100644
--- a/mediapipe/framework/calculator_graph.h
+++ b/mediapipe/framework/calculator_graph.h
@@ -53,10 +53,14 @@
 #include "mediapipe/framework/port/status.h"
 #include "mediapipe/framework/scheduler.h"
 #include "mediapipe/framework/thread_pool_executor.pb.h"
-#include "mediapipe/gpu/gpu_service.h"
 
 namespace mediapipe {
 
+#if !MEDIAPIPE_DISABLE_GPU
+class GpuResources;
+struct GpuSharedData;
+#endif  // !MEDIAPIPE_DISABLE_GPU
+
 typedef absl::StatusOr<OutputStreamPoller> StatusOrPoller;
 
 // The class representing a DAG of calculator nodes.