[XLA:CPU] Fix LinearOperatorPSDKernelTest.test_matmul_grad_xla_kernelparams_eager

Run the dot-decomposer after algebraic simplifier as the simplifier may introduce dot instructions which need decomposition. PiperOrigin-RevId: 294240419 Change-Id: Ib8f5ae7e88b2f1ddd569d72d4686185d3b3969e7
2020-02-10 09:47:24 -08:00 · 2020-02-10 09:47:24 -08:00 · 81cbebf768
commit 81cbebf768
parent cf35e6540f
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/compiler/xla/service/cpu/cpu_compiler.cc
+++ b/tensorflow/compiler/xla/service/cpu/cpu_compiler.cc
@ -310,6 +310,7 @@ Status CpuCompiler::RunHloPassesThroughLayoutAssn(
    pass.AddPass<HloConstantFolding>();
    pass.AddPass<ConditionalSimplifier>();
  }
+  pipeline.AddPass<DotDecomposer>();
  pipeline.AddPass<IndexedArrayAnalysisPrinterPass>();
  pipeline.AddPass<TransposeFolding>(
      [&](const HloInstruction& dot,