Added a sanity check

benoitsteiner · tensorflower-gardener · commit deb438298991 · 2017-05-12T00:13:01.000-07:00
PiperOrigin-RevId: 155840396
diff --git a/tensorflow/core/grappler/costs/virtual_placer.cc b/tensorflow/core/grappler/costs/virtual_placer.cc
@@ -25,8 +25,9 @@ namespace tensorflow {
 namespace grappler {
 
 VirtualPlacer::VirtualPlacer(const Cluster* cluster) : has_gpu_(false) {
+  CHECK(cluster);
   devices_ = cluster->GetDevices();
-  for (const auto& device : cluster->GetDevices()) {
+  for (const auto& device : devices_) {
     if (str_util::Lowercase(device.first).find("gpu") != string::npos) {
       has_gpu_ = true;
     }
