Częściowa ocena praworęcznych sekcji operatorów

Question

Mam funkcję (*~). Większość kosztów oceny x *~ y pochodzi z inspekcji drugi argument, z grubsza wzdłuż tych linii:

(*~) :: a -> b d -> c d a 
x *~ y = case y' of 
      Bar -> cheapFunction y' x 
      Baz -> cheapFunction2 y' x 
      Quux -> cheapFunction3 y' x 
    where 
    y' = expensive y 

Czy istnieje jakiś sposób, aby przekonać GHC częściowo ocenić odcinki operatora jak (*~ y)?

Próbowałem przepisanie go tak:

(*~) = flip go 
    where 
    go y = let y' = expensive y 
      in case y' of 
       Bar -> cheapFunction y' 
       Baz -> cheapFunction2 y' 
       Quux -> cheapFunction3 y' 

ale nie wydaje się, aby pomóc. Myślę, że może to być spowodowane tym, że flip wymaga wszystkich swoich argumentów przed wykonaniem przerzucania?

Jednym ze sposobów jest po prostu przerzucenie operatora, ale czyta się o wiele bardziej naturalnie, gdy kosztowny operand znajduje się po prawej stronie, ponieważ jest wyrównany z istniejącą notacją.

Czy mogę odpowiednio wypróbować {-# RULE #-}? Jeśli tak, co powinna ona powiedzieć? (Nie jestem pewien, jak daleko składnia będzie wyglądała, zanim reguły będą szukać dopasowań, między innymi).

Answer 1

Aby uruchomić taką optymalizację, musisz upewnić się, że twoja funkcja została zainspirowana. Przed deklaracją funkcji należy umieścić pragma {-# INLINE (*~) #-}. Nie mogę zagwarantować, że rozwiąże to twój problem, ale jest to jedyny sposób, w jaki widzę, że się do niego zbliża. Sprawdziłbym wygenerowany rdzeń kodu za pomocą narzędzia takiego jak "ghc-core", aby się upewnić.

Jednak twój problem jest tylko oznaką niewłaściwego składu kodu. Twoja funkcja wykonuje wiele niepowiązanych rzeczy. expensive y powinien po prostu być z tego zbudowany, wtedy twój problem zostanie usunięty jako taki. To jest wzór użycia powinien być x *~ expensive y zamiast x *~ y.