Add Video flag for add node of Video loaders (#117)

ROCm · Mar 28, 2024 · fa3df0e · fa3df0e
1 parent 5cb669b
commit fa3df0e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rocAL/include/pipeline/master_graph.h b/rocAL/include/pipeline/master_graph.h
@@ -351,6 +351,7 @@ inline std::shared_ptr<Cifar10LoaderNode> MasterGraph::add_node(const std::vecto
     return node;
 }
 
+#ifdef ROCAL_VIDEO
 /*
  * Explicit specialization for VideoLoaderNode
  */
@@ -389,3 +390,4 @@ inline std::shared_ptr<VideoLoaderSingleShardNode> MasterGraph::add_node(const s
 
     return node;
 }
+#endif