Add support of implicit second return value for div. Close #212

ryukzak · Jan 21, 2024 · 5e5d467 · 5e5d467
1 parent 0603636
commit 5e5d467
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/NITTA/Frontends/Lua.hs b/src/NITTA/Frontends/Lua.hs
@@ -428,6 +428,7 @@ alg2graph LuaAlgBuilder{algGraph, algLatestLuaValueInstance, algVars} = flip exe
         function2nitta LuaStatement{fName = "add", fIn = [a, b], fOut = [c], fValues = [], fInt = []} = F.add (fromText a) (fromText b) $ output c
         function2nitta LuaStatement{fName = "sub", fIn = [a, b], fOut = [c], fValues = [], fInt = []} = F.sub (fromText a) (fromText b) $ output c
         function2nitta LuaStatement{fName = "multiply", fIn = [a, b], fOut = [c], fValues = [], fInt = []} = F.multiply (fromText a) (fromText b) $ output c
+        function2nitta LuaStatement{fName = "divide", fIn = [d, n], fOut = [q], fValues = [], fInt = []} = F.division (fromText d) (fromText n) (output q) []
         function2nitta LuaStatement{fName = "divide", fIn = [d, n], fOut = [q, r], fValues = [], fInt = []} = F.division (fromText d) (fromText n) (output q) (output r)
         function2nitta LuaStatement{fName = "neg", fIn = [i], fOut = [o], fValues = [], fInt = []} = F.neg (fromText i) $ output o
         function2nitta LuaStatement{fName = "receive", fIn = [], fOut = [o], fValues = [], fInt = []} = F.receive $ output o

diff --git a/test/NITTA/Model/ProcessorUnits/Divider/Tests.hs b/test/NITTA/Model/ProcessorUnits/Divider/Tests.hs
@@ -137,14 +137,23 @@ tests =
             [ division "a" "b" ["c"] ["d"]
             ]
         , luaTestCase
-            "one division"
+            "one division explicit"
             [__i|
                 function f(a)
                     a, _b = a / 2
                     f(a)
                 end
                 f(1024)
             |]
+        , luaTestCase
+            "one division implicit"
+            [__i|
+                function f(a)
+                    a = a / 2
+                    f(a)
+                end
+                f(1024)
+            |]
         , luaTestCase
             "two division"
             [__i|