Added two missing changes during the porting

Author: jinlin-bayarea

src/cmd/compile/internal/inline/inl.go

@@ -267,10 +267,14 @@ func CanInline(fn *ir.Func) {
 		budget:        inlineMaxBudget,
 		extraCallCost: cc,
 	}
+	savefn := ir.CurFunc
+	ir.CurFunc = fn
 	if visitor.tooHairy(fn) {
 		reason = visitor.reason
+		ir.CurFunc = savefn
 		return
 	}
+	ir.CurFunc = savefn
 
 	n.Func.Inl = &ir.Inline{
 		Cost: inlineMaxBudget - visitor.budget,
@@ -581,10 +585,12 @@ func (v *hairyVisitor) doNode(n ir.Node) bool {
 
 	v.budget--
 
-	// When debugging, don't stop early, to get full cost of inlining this function
-	if v.budget < 0 && base.Flag.LowerM < 2 && !logopt.Enabled() {
-		v.reason = "too expensive"
-		return true
+	if pgo.WeightedCG == nil {
+		// When debugging, don't stop early, to get full cost of inlining this function
+		if v.budget < 0 && base.Flag.LowerM < 2 && !logopt.Enabled() {
+			v.reason = "too expensive"
+			return true
+		}
 	}
 
 	return ir.DoChildren(n, v.do)