Reverts, bug fix.

Signed-off-by: Michal Zientkiewicz <[email protected]>
NVIDIA · Jul 4, 2024 · fb6d729 · fb6d729
1 parent 665563c
commit fb6d729
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 6 deletions.
diff --git a/dali/operators/decoder/host/host_decoder.cc b/dali/operators/decoder/host/host_decoder.cc
@@ -22,7 +22,6 @@
 namespace dali {
 
 void HostDecoder::RunImpl(SampleWorkspace &ws) {
-  SetupSharedSampleParams(ws);
   const auto &input = ws.Input<CPUBackend>(0);
   auto &output = ws.Output<CPUBackend>(0);
   auto file_name = input.GetSourceInfo();

diff --git a/dali/operators/decoder/host/host_decoder.h b/dali/operators/decoder/host/host_decoder.h
@@ -1,4 +1,4 @@
-// Copyright (c) 2017-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// Copyright (c) 2017-2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -37,9 +37,6 @@ class HostDecoder : public StatelessOperator<CPUBackend> {
   DISABLE_COPY_MOVE_ASSIGN(HostDecoder);
 
  protected:
-  virtual void SetupSharedSampleParams(Workspace &ws) {}
-  using StatelessOperator<CPUBackend>::SetupSharedSampleParams;
-
   bool SetupImpl(std::vector<OutputDesc> &output_desc, const Workspace &ws) override {
     return false;
   }

diff --git a/dali/pipeline/executor/executor_impl.h b/dali/pipeline/executor/executor_impl.h
@@ -568,7 +568,7 @@ void Executor<WorkspacePolicy, QueuePolicy>::PruneUnusedGraphNodes() {
       if (!node.children.empty()) continue;
       // Do not prune the node if it has a preserve flag
       const auto &schema = node.spec.GetSchema();
-      if (node.spec.GetArgument<bool>("preserve") || !schema.IsNoPrune())
+      if (node.spec.GetArgument<bool>("preserve") || schema.IsNoPrune())
         continue;