Skip to content

Commit

Permalink
[XLA:GPU] Remove the call to FloatNormalization preceding the norma…
Browse files Browse the repository at this point in the history
…lization rewriter.

It should no longer be necessary. This is part of a series of changes aimed
at decreasing the slight added complexity introduced by the normalization
rewriter.

PiperOrigin-RevId: 707877388
  • Loading branch information
bchetioui authored and Google-ML-Automation committed Dec 19, 2024
1 parent 45597a0 commit bb49f2b
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions xla/service/gpu/gpu_compiler.cc
Original file line number Diff line number Diff line change
Expand Up @@ -1584,9 +1584,6 @@ absl::Status GpuCompiler::OptimizeHloPostLayoutAssignment(
if ((cuda_cc != nullptr &&
cuda_cc->IsAtLeast(se::CudaComputeCapability::AMPERE)) ||
rocm_cc != nullptr) {
// Triton compilation needs normalized operations on bf16 (i.e. converted
// to f32).
add_float_normalization(pipeline);
pipeline.AddPass<HloPassFix<GpuAlgebraicSimplifier>>(simplifier_options,
gpu_version);
pipeline.AddPass<HloCSE>(/*is_layout_sensitive=*/true);
Expand Down

0 comments on commit bb49f2b

Please sign in to comment.