Skip to content

Commit

Permalink
[XLA:CPU] Fix test failure for the cpu case on an internal test.
Browse files Browse the repository at this point in the history
The test fails because the simplifier does not accept F32_F32_F32 algorithm.

PiperOrigin-RevId: 700757953
  • Loading branch information
loislo authored and tensorflower-gardener committed Nov 27, 2024
1 parent d89833a commit 16bda38
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -3957,7 +3957,9 @@ absl::Status AlgebraicSimplifierVisitor::RewriteBatchPlusContractingAsReduce(

bool AlgebraicSimplifierVisitor::SupportedDotPrecisionConfig(
const PrecisionConfig& config) {
return config.algorithm() == PrecisionConfig::ALG_UNSET;
return config.algorithm() == PrecisionConfig::ALG_UNSET ||
// TODO(loislo): Fixes a failure on a test with CPU backend.
config.algorithm() == PrecisionConfig::ALG_DOT_F32_F32_F32;
}

absl::StatusOr<HloInstruction*>
Expand Down

0 comments on commit 16bda38

Please sign in to comment.