Skip to content

Commit 7ccafa5

Browse files
author
Frederik Mellbye
committed
Revert landing incorrect popdist::run diff
Summary: This reverts commit d004bc5. Test Plan: revert-hammer Reviewers: Subscribers:
1 parent d004bc5 commit 7ccafa5

File tree

3 files changed

+1
-10
lines changed

3 files changed

+1
-10
lines changed

tensorflow/compiler/plugin/poplar/driver/poplar_executor.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@ limitations under the License.
2525
#include <utility>
2626
#include <vector>
2727

28-
#include <popdist/backend.hpp>
2928
#include <poplar/DeviceManager.hpp>
3029
#include <poplar/IPUModel.hpp>
3130
#include <poplar/StreamCallback.hpp>
@@ -3789,7 +3788,7 @@ Status PoplarExecutor::ExecuteEngineImpl(se::DeviceMemoryBase* result_buffer,
37893788

37903789
// Run the main engine
37913790
current_engine_->enableExecutionProfiling();
3792-
popdist::run(*current_engine_, PoplarProgramType::MAIN_SEQUENCE);
3791+
current_engine_->run(PoplarProgramType::MAIN_SEQUENCE);
37933792

37943793
StopIOThreads();
37953794

tensorflow/compiler/plugin/poplar/tests/poprun_basic_test.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,6 @@
2828

2929

3030
class PoprunBasicTest(test_util.TensorFlowTestCase): # pylint: disable=abstract-method
31-
@classmethod
32-
def setUpClass(cls):
33-
popdist.init()
34-
3531
@tu.test_uses_ipus(num_ipus=4)
3632
@test_util.deprecated_graph_mode_only
3733
def test_cross_replica_sum(self):

tensorflow/python/ipu/tests/poprun_replica_partitioning_test.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -71,10 +71,6 @@ def cross_replica_reduce(self, grad):
7171

7272

7373
class PoprunReplicaPartitioningTest(test.TestCase):
74-
@classmethod
75-
def setUpClass(cls):
76-
popdist.init()
77-
7874
def _compare_partitioned_to_non_partitioned(self, stages, repeat_count,
7975
gradient_accumulation_count,
8076
dataset_fn, optimizer_fn):

0 commit comments

Comments
 (0)