Fix cholesky

ROCm · Oct 31, 2024 · 15f2f3e · 15f2f3e
1 parent e373633
commit 15f2f3e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/xla/service/gpu/ir_emitter_unnested.cc b/xla/service/gpu/ir_emitter_unnested.cc
@@ -2898,6 +2898,9 @@ absl::Status IrEmitterUnnested::EmitHloInstruction(
       if (instr->custom_call_target() == kNopCustomCallTarget) {
         return absl::OkStatus();
       }
+      if (IsCustomCallToCusolver(*instr)) {
+        return EmitCholeskyThunk(instr);
+      }
       return EmitCustomCallThunk(custom_call);
     }
     case HloOpcode::kFusion: {