Allow reusing temporary object files generated by GHCi by writing to -odir in case -fwrite-interface was specified (#16670)

8 files changed, 60 insertions, 5 deletions

diff --git a/compiler/main/GhcMake.hs b/compiler/main/GhcMake.hs
index 8767a6e99c..81311c1e0c 100644
--- a/compiler/main/GhcMake.hs
+++ b/compiler/main/GhcMake.hs
@@ -2198,15 +2198,17 @@ enableCodeGenWhen condition should_modify staticLife dynLife target nodemap =
           -- to by the user. But we need them, so we patch their locations in
           -- the ModSummary with temporary files.
           --
-        hi_file <-
+        (hi_file, o_file) <-
+          -- If ``-fwrite-interface` is specified, then the .o and .hi files
+          -- are written into `-odir` and `-hidir` respectively.  #16670
           if gopt Opt_WriteInterface dflags
-            then return $ ml_hi_file ms_location
-            else new_temp_file (hiSuf dflags) (dynHiSuf dflags)
-        o_temp_file <- new_temp_file (objectSuf dflags) (dynObjectSuf dflags)
+            then return (ml_hi_file ms_location, ml_obj_file ms_location)
+            else (,) <$> (new_temp_file (hiSuf dflags) (dynHiSuf dflags))
+                     <*> (new_temp_file (objectSuf dflags) (dynObjectSuf dflags))
         return $
           ms
           { ms_location =
-              ms_location {ml_hi_file = hi_file, ml_obj_file = o_temp_file}
+              ms_location {ml_hi_file = hi_file, ml_obj_file = o_file}
           , ms_hspp_opts = updOptLevel 0 $ dflags {hscTarget = target}
           }
       | otherwise = return ms
diff --git a/testsuite/tests/ghci/T16670/Makefile b/testsuite/tests/ghci/T16670/Makefile
new file mode 100644
index 0000000000..586f7e7fe7
--- /dev/null
+++ b/testsuite/tests/ghci/T16670/Makefile
@@ -0,0 +1,24 @@
+TOP=../../..
+include $(TOP)/mk/boilerplate.mk
+include $(TOP)/mk/test.mk
+
+clean:
+	$(RM) -rf my-odir
+
+T16670_unboxed:
+	@echo "~~~~~~~~ testing T16670_unboxed"
+	$(MAKE) -s --no-print-directory clean
+	mkdir my-odir
+	echo ":load T16670_unboxed.hs" | "$(TEST_HC)" $(TEST_HC_OPTS_INTERACTIVE) \
+		-v0 -fwrite-interface -odir my-odir
+	find . -name T16670_unboxed.o
+	test -f my-odir/T16670_unboxed.o
+
+T16670_th:
+	@echo "~~~~~~~~ testing T16670_th"
+	$(MAKE) -s --no-print-directory clean
+	mkdir my-odir
+	echo ":load T16670_th.hs" | "$(TEST_HC)" $(TEST_HC_OPTS_INTERACTIVE) \
+		-v0 -fno-code -fwrite-interface -odir my-odir
+	find . -name T16670_th.o
+	test -f my-odir/T16670_th.o
diff --git a/testsuite/tests/ghci/T16670/T16670_th.hs b/testsuite/tests/ghci/T16670/T16670_th.hs
new file mode 100644
index 0000000000..f21c7991f1
--- /dev/null
+++ b/testsuite/tests/ghci/T16670/T16670_th.hs
@@ -0,0 +1,7 @@
+{-# LANGUAGE TemplateHaskell #-}
+
+module T16670_th where
+
+import TH
+
+x = $(th)
diff --git a/testsuite/tests/ghci/T16670/T16670_th.stdout b/testsuite/tests/ghci/T16670/T16670_th.stdout
new file mode 100644
index 0000000000..708b93a384
--- /dev/null
+++ b/testsuite/tests/ghci/T16670/T16670_th.stdout
@@ -0,0 +1,2 @@
+~~~~~~~~ testing T16670_th
+./my-odir/T16670_th.o
diff --git a/testsuite/tests/ghci/T16670/T16670_unboxed.hs b/testsuite/tests/ghci/T16670/T16670_unboxed.hs
new file mode 100644
index 0000000000..2e903959bb
--- /dev/null
+++ b/testsuite/tests/ghci/T16670/T16670_unboxed.hs
@@ -0,0 +1,8 @@
+{-# LANGUAGE UnboxedTuples #-}
+{-# OPTIONS_GHC -fwrite-interface #-}
+module T16670_unboxed where
+
+data UnboxedTupleData = MkUTD (# (),() #)
+
+doThings :: UnboxedTupleData -> ()
+doThings (MkUTD t) = ()
diff --git a/testsuite/tests/ghci/T16670/T16670_unboxed.stdout b/testsuite/tests/ghci/T16670/T16670_unboxed.stdout
new file mode 100644
index 0000000000..61c74ca01d
--- /dev/null
+++ b/testsuite/tests/ghci/T16670/T16670_unboxed.stdout
@@ -0,0 +1,2 @@
+~~~~~~~~ testing T16670_unboxed
+./my-odir/T16670_unboxed.o
diff --git a/testsuite/tests/ghci/T16670/TH.hs b/testsuite/tests/ghci/T16670/TH.hs
new file mode 100644
index 0000000000..f288c784f0
--- /dev/null
+++ b/testsuite/tests/ghci/T16670/TH.hs
@@ -0,0 +1,3 @@
+{-# LANGUAGE TemplateHaskell #-}
+module TH where
+th = [|909|]
diff --git a/testsuite/tests/ghci/T16670/all.T b/testsuite/tests/ghci/T16670/all.T
new file mode 100644
index 0000000000..00a4f5311c
--- /dev/null
+++ b/testsuite/tests/ghci/T16670/all.T
@@ -0,0 +1,7 @@
+test('T16670_unboxed', [extra_files(['T16670_unboxed.hs']),
+    req_interp, omit_ways(prof_ways), unless(have_dynamic(), skip)],
+        makefile_test, ['T16670_unboxed'])
+
+test('T16670_th', [extra_files(['T16670_th.hs', 'TH.hs']),
+    req_interp, omit_ways(prof_ways), unless(have_dynamic(), skip)],
+        makefile_test, ['T16670_th'])