diff options
author | 2018-04-12 10:14:02 -0700 | |
---|---|---|
committer | 2018-04-12 10:16:34 -0700 | |
commit | ffbf77de81d0b7b4b169c92d0d9fbbdef5b8842a (patch) | |
tree | 8d68eedf28bdcac55516b6a3a176d56f6cef0fa2 /tensorflow/compiler/xla/tests/test_utils.cc | |
parent | 8a247976484173059aedc17bfd8d770b8d1a70e1 (diff) |
Introduced tool to run an HLO module in replicated fashion, by infeeding random data and outfeeding the data generated at each step.
The arguments of the computation can be either read from the session module, or randomly generated.
The tool uses the raw transfer manager API to infeed and outfeed the data.
PiperOrigin-RevId: 192628605
Diffstat (limited to 'tensorflow/compiler/xla/tests/test_utils.cc')
-rw-r--r-- | tensorflow/compiler/xla/tests/test_utils.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/tests/test_utils.cc b/tensorflow/compiler/xla/tests/test_utils.cc index e30d115fae..cda1989fad 100644 --- a/tensorflow/compiler/xla/tests/test_utils.cc +++ b/tensorflow/compiler/xla/tests/test_utils.cc @@ -340,8 +340,8 @@ StatusOr<std::vector<std::unique_ptr<Literal>>> MakeFakeArguments( } Status VerifyHloModule(const perftools::gputools::Platform& platform, - HloModule* const module) { - return HloVerifier().Run(module).status(); + HloModule* const module, bool allow_mixed_precision) { + return HloVerifier(allow_mixed_precision).Run(module).status(); } } // namespace xla |