From 81cbebf768c2340b3d334b6232dc2db09d8cdbe8 Mon Sep 17 00:00:00 2001
From: David Majnemer <majnemer@google.com>
Date: Mon, 10 Feb 2020 09:47:24 -0800
Subject: [PATCH] [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
---
tensorflow/compiler/xla/service/cpu/cpu_compiler.cc | 1 +
1 file changed, 1 insertion(+)
diff --git a/tensorflow/compiler/xla/service/cpu/cpu_compiler.cc b/tensorflow/compiler/xla/service/cpu/cpu_compiler.cc
index 8fbe29f417c..446179c217f 100644
--- 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,