diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-22 05:15:18 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-22 05:19:06 -0700 |
| commit | ca552d54ac67be8837aeabdb43269846d9df4eb5 (patch) | |
| tree | 11d592685766ab64187b520d91d7dfa2b6f231fc /tensorflow/core/common_runtime | |
| parent | e317152dad1aa66bc493abc046a60dbbf650de92 (diff) | |
Add PinToHostOptimizer to grappler: force small ops to happen on CPU (instead of
GPU). This avoids many unnecessary CPU<->GPU memcpy and syncs.
PiperOrigin-RevId: 214108484
Diffstat (limited to 'tensorflow/core/common_runtime')
| -rw-r--r-- | tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc b/tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc index 0b096a14a3..2ed4f69f90 100644 --- a/tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc +++ b/tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc @@ -77,6 +77,9 @@ TEST(DirectSessionWithTrackingAllocTest, CostModelTest) { options.config.mutable_graph_options() ->mutable_rewrite_options() ->set_min_graph_nodes(-1); + options.config.mutable_graph_options() + ->mutable_rewrite_options() + ->set_pin_to_host_optimization(RewriterConfig::OFF); std::unique_ptr<Session> session(NewSession(options)); TF_ASSERT_OK(session->Create(def)); std::vector<std::pair<string, Tensor>> inputs; |