From 68484bdc68246dc896811e767fed6ca6bcd5e1bd Mon Sep 17 00:00:00 2001
From: Josh Williams <48445365+jvwilliams23@users.noreply.github.com>
Date: Mon, 22 Jul 2024 15:19:25 +0100
Subject: [PATCH] fix missing header for dump_model_graph (#2464)

fixes "invalid use of incomplete type lbann::model" when compiling with LBANN_HAS_BOOST
---
 src/callbacks/dump_model_graph.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/callbacks/dump_model_graph.cpp b/src/callbacks/dump_model_graph.cpp
index 890cb8f5980..f4584f2f35d 100644
--- a/src/callbacks/dump_model_graph.cpp
+++ b/src/callbacks/dump_model_graph.cpp
@@ -25,6 +25,7 @@
 ////////////////////////////////////////////////////////////////////////////////
 
 #include "lbann/callbacks/dump_model_graph.hpp"
+#include "lbann/models/model.hpp"
 #include "lbann/utils/exception.hpp"
 #include "lbann_config.hpp"