Clean up some code after previous CL

PiperOrigin-RevId: 158282834
This commit is contained in:
Eli Bendersky 2017-06-07 10:09:52 -07:00 committed by TensorFlower Gardener
parent 7b5302af0a
commit dffea202a5
4 changed files with 7 additions and 8 deletions

View File

@ -64,8 +64,7 @@ StatusOr<bool> HloPassPipeline::Run(HloModule* module) {
bool changed = false; bool changed = false;
string message; string message;
for (auto& pass : passes_) { for (auto& pass : passes_) {
if (!disabled_passes.empty() && if (disabled_passes.count(pass->name().ToString()) > 0) {
disabled_passes.count(pass->name().ToString()) > 0) {
VLOG(1) << " Skipping HLO pass " << pass->name() VLOG(1) << " Skipping HLO pass " << pass->name()
<< ", disabled by --xla_disable_hlo_passes"; << ", disabled by --xla_disable_hlo_passes";
continue; continue;

View File

@ -38,8 +38,8 @@ class ConvertTest : public ClientLibraryTestBase {
public: public:
explicit ConvertTest(perftools::gputools::Platform* platform = nullptr) explicit ConvertTest(perftools::gputools::Platform* platform = nullptr)
: ClientLibraryTestBase(platform) { : ClientLibraryTestBase(platform) {
mutable_debug_options()->mutable_xla_disable_hlo_passes()->Add("algsimp"); mutable_debug_options()->add_xla_disable_hlo_passes("algsimp");
mutable_debug_options()->mutable_xla_disable_hlo_passes()->Add("inline"); mutable_debug_options()->add_xla_disable_hlo_passes("inline");
} }
}; };

View File

@ -43,8 +43,8 @@ class MapTest : public ClientLibraryTestBase {
public: public:
explicit MapTest(perftools::gputools::Platform* platform = nullptr) explicit MapTest(perftools::gputools::Platform* platform = nullptr)
: ClientLibraryTestBase(platform) { : ClientLibraryTestBase(platform) {
mutable_debug_options()->mutable_xla_disable_hlo_passes()->Add("algsimp"); mutable_debug_options()->add_xla_disable_hlo_passes("algsimp");
mutable_debug_options()->mutable_xla_disable_hlo_passes()->Add("inline"); mutable_debug_options()->add_xla_disable_hlo_passes("inline");
} }
// Creates a function that adds its scalar argument with the constant 1.0. // Creates a function that adds its scalar argument with the constant 1.0.

View File

@ -43,8 +43,8 @@ class VecOpsSimpleTest : public ClientLibraryTestBase {
public: public:
explicit VecOpsSimpleTest(perftools::gputools::Platform* platform = nullptr) explicit VecOpsSimpleTest(perftools::gputools::Platform* platform = nullptr)
: ClientLibraryTestBase(platform) { : ClientLibraryTestBase(platform) {
mutable_debug_options()->mutable_xla_disable_hlo_passes()->Add("algsimp"); mutable_debug_options()->add_xla_disable_hlo_passes("algsimp");
mutable_debug_options()->mutable_xla_disable_hlo_passes()->Add("inline"); mutable_debug_options()->add_xla_disable_hlo_passes("inline");
} }
ErrorSpec error_spec_{0.0001}; ErrorSpec error_spec_{0.0001};