Can/does the (forward) pipe operator prevent tail call optimization?

后端 未结 1 1619
我寻月下人不归
我寻月下人不归 2020-12-18 20:12

For a parameter optimization problem at work I wrote a genetic algorithm to find some good settings because a brute-force solution is unfeasible. Unfortunately, when I retur

相关标签:
1条回答
  • 2020-12-18 20:29

    Based on the minimal case as provided, if the code is run in release mode in 64-bit it fails with a stack overflow. If the code is run in release mode in 32-bit mode it succeeds.

    Note: The option to choose between 32-bit and 64-bit is Prefer 32-bit as seen in the images below.

    Increasing the stack size will result in the code succeeding in release mode in 64-bit. This is done via the use of the Thread constructor.

    [<EntryPoint>]
    let main _ =
    
        let test () =
            let r = KissRandom()
            let n = r.Normal()
            Seq.item 20000 n |> printfn "%f"
    
        /// The greatest maximum-stack-size that should be used
        /// with the 'runWithStackFrame' function.
        let STACK_LIMIT = 16777216
    
        /// Run a function with a custom maximum stack size.
        /// This is necessary for some functions to execute
        /// without raising a StackOverflowException.
        let runWithCustomStackSize maxStackSize fn =
            // Preconditions
            if maxStackSize < 1048576 then
                invalidArg "stackSize" "Functions should not be executed with a \
                    maximum stack size of less than 1048576 bytes (1MB)."
            elif maxStackSize > STACK_LIMIT then
                invalidArg "stackSize" "The maximum size of the stack frame should \
                    not exceed 16777216 bytes (16MB)."
    
            /// Holds the return value of the function.
            let result = ref Unchecked.defaultof<'T>
    
            // Create a thread with the specified maximum stack size,
            // then immediately execute the function on it.
            let thread = System.Threading.Thread ((fun () -> result := fn()), maxStackSize)
            thread.Start ()
    
            // Wait for the function/thread to finish and return the result.
            thread.Join ()
            !result
    
        /// Runs a function within a thread which has an enlarged maximum-stack-size.
        let inline runWithEnlargedStack fn =
            runWithCustomStackSize STACK_LIMIT fn
    
    
    //    test ()       // Fails with stack overflow in 64-bit mode, Release
                        // Runs successfully in 32-bit mode, Release
    
        runWithEnlargedStack test
        
        printf "Press any key to exit: "
        System.Console.ReadKey() |> ignore
        printfn ""
    
        0
    

    This code is from FSharp-logic-examples and in particular Anh-Dung Phan

    While I have not checked for the root cause, I suspect it is because of the size of the items for 64-bit is larger that the size of the items for 32-bit and even though the number of items put into the stack and the stack size remains the same for both versions, the item size increase pushes the memory needed for the stack over the 1 megabyte limit.

    TL;DR

    This has been a fun and enlightening question to answer. I am glad it was asked.

    Originally the problem appeared to be related to the use of |> and TCO and since that is still of value I am leaving it in the answer. I would also like to thank the OP for being response and helpful, it is a pleasure to help someone who works with you and not against you.

    In the following code which is recursive and has |> is run in Debug mode within Visual Studio it causes a StackOverflow.

    If it is started from the command line from the bin\release directory it does NOT cause a StackOverflow.

    Using Visual Studio 15 Community

    [<EntryPoint>]
    let main argv = 
    
        let largeList = 
            printfn "Creating large list"
            [
                for i in 1 .. 100000000 do
                    yield i
            ]
    
        // causes StackOverflow in Debug
        // No StackOverflow in Release
        let sum4 l =
            printfn "testing sum4"
            let rec sumInner4 l acc =
                match l with
                | h::t -> 
                    let acc = acc + h
                    acc |> sumInner4 t
                | [] -> acc
            sumInner4 l 0
    
        let result4 = sum4 largeList
        printfn "result4: %A" result4
    

    Where Release or Debug is set in the Visual Studio toolbar

    and the options for the project in Debug mode are

    and the options for the project in Release mode are

    tldr;

    In the process of testing this I created 16 different test and built them in both debug and release mode and verified if they ran to completion or threw a stack overflow. The 16 are broken down into a set of 4 with 4 cases each. The cases 1,5,9,13 are a negative and produce a stack overflow to ensure that a stack overflow can be created. Cases 2,6,10,14 are a positive to show that the tail call is working and not causing a stack overflow. Cases 3,7,11,15 show a tail call with an operation done in the same statement as the tail call, and to be one factorization away from the test cases using |>; these work as expected. Cases 4,8,12,16 use |> and shows when it does and does not work in debug mode, which is probably a surprise to many. Cases 1-4 and 9-12 use a function of the form f x y, cases 8-11 use a function of the form f x and cases 12-16 use a function of the form f x y z. I originally did the first 8 test cases but after Keith's comment did 4 more which don't use a list but still use a function of the from f x y and present the unexpected result and then did 4 more that use a function of the form f x y z.

    To run a test you will have to comment out all but the one test you plan to run and the build it once in debug mode, which can then be run from within Visual Studio, and then again build it in release mode and run it. I run it from a command line to ensure I am running the release version.

    [<EntryPoint>]
    let main argv = 
    
        let largeList = 
            printfn "Creating large list"
            [
                for i in 1 .. 100000000 do
                    yield i
            ]
    
        // causes StackOverflow in Debug
        // causes StackOverflow in Release
        //   Negative confirmation
        //   A supposed tail call that DOES cause a stack overflow in both debug and release mode
        //   options: f x y
        let sum1 l = 
            printfn "test 01: "
            let rec sum1Inner l acc =
                match l with
                | h::t -> 
                    let acc = acc + h
                    1 + sum1Inner t acc
                | [] -> acc
            sum1Inner l 0
            
        // No StackOverflow in Debug
        // No StackOverflow in Release
        //   Positive confirmation
        //   A tail call that DOES NOT cause a stack overflow in both debug and release mode
        //   options: f x y
        let sum2 l =
            printfn "test 02: "
            let rec sum2Inner l acc =
                match l with
                | h::t -> 
                    let acc = acc + h
                    sum2Inner t acc
                | [] -> acc
            sum2Inner l 0
            
        // No StackOverflow in Debug
        // No StackOverflow in Release
        //   A test case
        //   options: f x y and no |>
        let sum3 l =
            printfn "test 03: "
            let rec sum3Inner l acc =
                match l with
                | h::t -> 
                    sum3Inner t (acc + h)
                | [] -> acc
            sum3Inner l 0
            
        // causes StackOverflow in Debug
        // No StackOverflow in Release
        //   A test case
        //   options: f x y and |>
        let sum4 l =
            printfn "test 04: "
            let rec sum4Inner l acc =
                match l with
                | h::t -> 
                    let acc = acc + h
                    acc |> sum4Inner t
                | [] -> acc
            sum4Inner l 0
            
        // causes StackOverflow in Debug
        // causes StackOverflow in Release
        //   Negative confirmation
        //   A supposed tail call that DOES cause a stack overflow in both debug and release mode
        //   options: f x
        let sum5 () =
            printfn "test 05: "
            let rec sum5Inner x =
                match x with 
                | 10000000 -> x
                | _ -> 
                    let acc = x + 1
                    1 + sum5Inner acc
            sum5Inner 0
            
        // No StackOverflow in Debug
        // No StackOverflow in Release
        //   Positive confirmation
        //   A tail call that DOES NOT cause a stack overflow in both debug and release mode
        //   options: f x
        let sum6 () =
            printfn "test 06: "
            let rec sum6Inner x =
                match x with 
                | 10000000 -> x
                | _ -> 
                    let acc = x + 1
                    sum6Inner acc
            sum6Inner 0
            
        // No StackOverflow in Debug
        // No StackOverflow in Release
        //  A test case
        //  options: f x and no |>
        let sum7 l =
            printfn "test 07: "
            let rec sum7Inner x =
                match x with 
                | 10000000 -> x
                | _ -> sum7Inner (x + 1)
            sum7Inner 0
            
        // No StackOverflow in Debug
        // No StackOverflow in Release
        //   A test case
        //   options: f x and |>
        let sum8 () =
            printfn "test 07: "
            let rec sumInner8 x =
                match x with
                | 10000000 -> x
                | _ -> 
                    let acc = x + 1
                    acc |> sumInner8 
            sumInner8 0
    
        // causes StackOverflow in Debug
        // causes StackOverflow in Release
        //   Negative confirmation"
        //   A supposed tail call that DOES cause a stack overflow in both debug and release mode"
        //   options: f x y"
        let sum9 () = 
            printfn "test 09: "
            let rec sum9Inner x y =
                match y with
                | 10000000 -> y
                | _ -> 
                    let acc = x + y
                    1 + sum9Inner x acc
            sum9Inner 1 0   
            
        // No StackOverflow in Debug
        // No StackOverflow in Release
        //   Positive confirmation
        //   A tail call that DOES NOT cause a stack overflow in both debug and release mode
        //   options: f x y
        let sum10 () =
            printfn "test 10: "
            let rec sum10Inner x y =
                match y with
                | 10000000 -> y
                | _ -> 
                    let acc = x + y
                    sum10Inner x acc
            sum10Inner 1 0
    
        // No StackOverflow in Debug
        // No StackOverflow in Release
        //   A test case
        //   options: f x y and no |>
        let sum11 () =
            printfn "test 11: "
            let rec sum11Inner x y =
                match y with
                | 10000000 -> y
                | _ -> 
                    sum11Inner x (x + y) 
            sum11Inner 1 0
            
        // causes StackOverflow in Debug
        // No StackOverflow in Release
        //   A test case
        //   options: f x y and |>
        let sum12 () =
            printfn "test 12: "
            let rec sum12Inner x y =
                match y with
                | 10000000 -> y
                | _ -> 
                    let acc = x + y
                    acc |> sum12Inner x
            sum12Inner 1 0
    
        // causes StackOverflow in Debug
        // No StackOverflow in Release
        //   A test case"
        //   options: f x y and |>"
        let sum12 () =
            printfn "test 12: "
            let rec sum12Inner x y =
                match y with
                | 10000000 -> y
                | _ -> 
                    let acc = x + y
                    acc |> sum12Inner x
            sum12Inner 1 0
    
        // causes StackOverflow in Debug
        // causes StackOverflow in Release
        //   Negative confirmation"
        //   A supposed tail call that DOES cause a stack overflow in both debug and release mode"
        //   options: f x y"
        let sum13 () = 
            printfn "test 13: "
            let rec sum13Inner x z y =
                match y with
                | 10000000 -> y
                | _ -> 
                    let acc = x + y
                    1 + sum13Inner x z acc 
            sum13Inner 1 "z" 0
            
        // No StackOverflow in Debug
        // No StackOverflow in Release
        //   Positive confirmation"
        //   A tail call that DOES NOT cause a stack overflow in both debug and release mode"
        //   options: f x y"
        let sum14 () =
            printfn "test 14: "
            let rec sum14Inner x z y =
                match y with
                | 10000000 -> y
                | _ -> 
                    let acc = x + y
                    sum14Inner x z acc
            sum14Inner 1 "z" 0
    
        // No StackOverflow in Debug
        // No StackOverflow in Release
        //   A test case"
        //   options: f x y and no |>"
        let sum15 () =
            printfn "test 15: "
            let rec sum15Inner x z y =
                match y with
                | 10000000 -> y
                | _ -> 
                    sum15Inner x z (x + y) 
            sum15Inner 1 "z" 0
    
        // causes StackOverflow in Debug
        // No StackOverflow in Release
        //   A test case"
        //   options: f x y and |>"
        let sum16 () =
            printfn "test 16: "
            let rec sum16Inner x z y =
                match y with
                | 10000000 -> y
                | _ -> 
                    let acc = x + y
                    acc |> sum16Inner x z
            sum16Inner 1 "z" 0
    
        let result1 = sum1 largeList
        printfn "result1: %A" result1
    
        let result2 = sum2 largeList
        printfn "result2: %A" result2
    
        let result3 = sum3 largeList
        printfn "result3: %A" result3
    
        let result4 = sum4 largeList
        printfn "result4: %A" result4
    
        let result5 = sum5 ()
        printfn "result5: %A" result5
    
        let result6 = sum6 ()
        printfn "result6: %A" result6
    
        let result7 = sum7 ()
        printfn "result7: %A" result7
    
        let result8 = sum8 ()
        printfn "result8: %A" result8
    
        let result9 = sum9 ()
        printfn "result9: %A" result9
    
        let result10 = sum10 ()
        printfn "result10: %A" result10
    
        let result11 = sum11 ()
        printfn "result11: %A" result11
    
        let result12 = sum12 ()
        printfn "result12: %A" result12
    
        let result13 = sum13 ()
        printfn "result13: %A" result13
    
        let result14 = sum14 ()
        printfn "result14: %A" result14
    
        let result15 = sum15 ()
        printfn "result15: %A" result15
    
        let result16 = sum16 ()
        printfn "result16: %A" result16
        
        printf "Press any key to exit: "
        System.Console.ReadKey() |> ignore
        printfn ""
    
        0 // return an integer exit code
    

    Additional, new info

    EDIT: This thread on Github has Don Syme, creator of F#, specifically mention that:

    [...] Second, you are correct, we don't guarantee to optimize uses of f <| x or x |> f or any similar to first-calling tailcalls even if f x is a tailcall.

    0 讨论(0)
提交回复
热议问题