Compile: special-case to a single kernel functionHEADmaster

1 files changed, 8 insertions, 11 deletions

diff --git a/src/Compile/Exec.hs b/src/Compile/Exec.hs
index d708fc0..9b9fb15 100644
--- a/src/Compile/Exec.hs
+++ b/src/Compile/Exec.hs
@@ -11,8 +11,6 @@ module Compile.Exec (
 
 import Control.Monad (when)
 import Data.IORef
-import qualified Data.Map.Strict as Map
-import Data.Map.Strict (Map)
 import Foreign (Ptr)
 import Foreign.Ptr (FunPtr)
 import System.Directory (removeDirectoryRecursive)
@@ -30,10 +28,10 @@ debug :: Bool
 debug = False
 
 -- The IORef wrapper is required for the finalizer to attach properly (see the 'Weak' docs)
-data KernelLib = KernelLib !(IORef (Map String (FunPtr (Ptr () -> IO ()))))
+data KernelLib = KernelLib !(IORef (FunPtr (Ptr () -> IO ())))
 
-buildKernel :: String -> [String] -> IO KernelLib
-buildKernel csource funnames = do
+buildKernel :: String -> String -> IO KernelLib
+buildKernel csource funname = do
   template <- (++ "/tmp.chad.") <$> getTempDir
   path <- mkdtemp template
 
@@ -69,8 +67,7 @@ buildKernel csource funnames = do
 
   removeDirectoryRecursive path  -- we keep a reference anyway because we have the file open now
 
-  ptrs <- Map.fromList <$> sequence [(name,) <$> dlsym dl name | name <- funnames]
-  ref <- newIORef ptrs
+  ref <- newIORef =<< dlsym dl funname
   _ <- mkWeakIORef ref (do numLeft <- atomicModifyIORef' numLoadedCounter (\n -> (n-1, n-1))
                            when debug $ hPutStrLn stderr $ "[chad] unloading kernel " ++ path ++ " (" ++ show numLeft ++ " left)"
                            dlclose dl)
@@ -81,10 +78,10 @@ foreign import ccall "dynamic"
 
 -- Ensure that keeping a reference to the returned function also keeps the 'KernelLib' alive
 {-# NOINLINE callKernelFun #-}
-callKernelFun :: String -> KernelLib -> Ptr () -> IO ()
-callKernelFun key (KernelLib ref) arg = do
-  mp <- readIORef ref
-  wrapKernelFun (mp Map.! key) arg
+callKernelFun :: KernelLib -> Ptr () -> IO ()
+callKernelFun (KernelLib ref) arg = do
+  ptr <- readIORef ref
+  wrapKernelFun ptr arg
 
 getTempDir :: IO FilePath
 getTempDir =