fix: use correct return

Author: avik-pal

src/enzyme_ad/jax/Passes/LowerEnzymeXLALinalg.cpp

@@ -108,7 +108,7 @@ struct LUFactorizationOpLowering
         /*output_operand_aliases=*/outputOperandAliases,
         /*xla_side_effect_free=*/rewriter.getUnitAttr());
 
-    stablehlo::ReturnOp::create(rewriter, op.getLoc(),
+    func::ReturnOp::create(rewriter, op.getLoc(),
                                 ValueRange{jitCall.getResult(0),
                                            jitCall.getResult(1),
                                            jitCall.getResult(2)});