diff --git a/mediapipe/framework/calculator_graph.h b/mediapipe/framework/calculator_graph.h
index 04f9de45f..8d58ff312 100644
--- a/mediapipe/framework/calculator_graph.h
+++ b/mediapipe/framework/calculator_graph.h
@@ -53,14 +53,10 @@
 #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.