Commit b7c8dab7 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Merge pull request #7439 from tfboyd:lstm_no_xla_8_gpu

PiperOrigin-RevId: 263261200
parents c789d64b 0d92f34a
...@@ -208,21 +208,6 @@ class ShakespeareAccuracy(ShakespeareBenchmarkBase): ...@@ -208,21 +208,6 @@ class ShakespeareAccuracy(ShakespeareBenchmarkBase):
FLAGS.model_dir = '' FLAGS.model_dir = ''
self._run_and_report_benchmark() self._run_and_report_benchmark()
def benchmark_xla_8_gpu(self):
"""Benchmark 8 gpu w/xla.
This is test is for accuracy not scaling. The batch-size is not scaled to
the number of gpus.
"""
self._setup()
FLAGS.num_gpus = 8
FLAGS.training_data = self.train_data
FLAGS.batch_size = 64
FLAGS.train_epochs = 43
FLAGS.model_dir = ''
FLAGS.enable_xla = True
self._run_and_report_benchmark()
class ShakespeareKerasBenchmarkReal(ShakespeareBenchmarkBase): class ShakespeareKerasBenchmarkReal(ShakespeareBenchmarkBase):
"""Benchmark accuracy tests.""" """Benchmark accuracy tests."""
......
...@@ -26,7 +26,7 @@ limitations under the License. ...@@ -26,7 +26,7 @@ limitations under the License.
#include "mobile_ssd_client.h" #include "mobile_ssd_client.h"
#include "protos/anchor_generation_options.pb.h" #include "protos/anchor_generation_options.pb.h"
#ifdef ENABLE_EDGETPU #ifdef ENABLE_EDGETPU
#include "libedgetpu/edgetpu.h" #include "libedgetpu/libedgetpu.h"
#endif // ENABLE_EDGETPU #endif // ENABLE_EDGETPU
namespace lstm_object_detection { namespace lstm_object_detection {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment