Merge pull request #1587 from certik/floordiv

Implement floordiv for i16 and i8

Author: certik

integration_tests/CMakeLists.txt

@@ -255,6 +255,7 @@ RUN(NAME expr_12             LABELS llvm c)
 RUN(NAME expr_13             LABELS llvm c
         EXTRAFILES expr_13b.c)
 RUN(NAME expr_14             LABELS cpython llvm c)
+RUN(NAME expr_15             LABELS cpython llvm c)
 RUN(NAME loop_01             LABELS cpython llvm c)
 RUN(NAME loop_02             LABELS cpython llvm c wasm wasm_x86 wasm_x64)
 RUN(NAME loop_03             LABELS cpython llvm c wasm wasm_x64)

integration_tests/expr_15.py

@@ -0,0 +1,19 @@
+from ltypes import i8, i16
+
+def test_issue_1586():
+    x4: i16
+    y4: i16
+    y4 = i16(10)
+    x4 = i16(12345)
+    assert x4//y4 == i16(1234)
+
+    a4: i8
+    b4: i8
+    a4 = i8(10)
+    b4 = i8(123)
+    assert b4//a4 == i8(12)
+
+def check():
+    test_issue_1586()
+
+check()

src/runtime/lpython_builtin.py

@@ -469,6 +469,26 @@ def _lpython_floordiv(a: f32, b: f32) -> f32:
         resultf32 = f32(1.0) * f32(result) - f32(1.0)
     return resultf32
 
+@overload
+def _lpython_floordiv(a: i8, b: i8) -> i8:
+    r: f64 # f32 rounds things up and gives incorrect results
+    r = float(a)/float(b)
+    result: i8
+    result = i8(r)
+    if r >= 0.0 or f64(result) == r:
+        return result
+    return result - i8(1)
+
+@overload
+def _lpython_floordiv(a: i16, b: i16) -> i16:
+    r: f64 # f32 rounds things up and gives incorrect results
+    r = float(a)/float(b)
+    result: i16
+    result = i16(r)
+    if r >= 0.0 or f64(result) == r:
+        return result
+    return result - i16(1)
+
 @overload
 def _lpython_floordiv(a: i32, b: i32) -> i32:
     r: f64 # f32 rounds things up and gives incorrect results

tests/reference/asr-array_01_decl-39cf894.json

@@ -6,7 +6,7 @@
     "outfile": null,
     "outfile_hash": null,
     "stdout": "asr-array_01_decl-39cf894.stdout",
-    "stdout_hash": "9a5b7daf71fa50d535cc4ed81bfbc307ffb5e0560317e8e08de0e70d",
+    "stdout_hash": "5ecd298c183e6b4027f8f601a2b5d3963033ebf54bca582c982b5929",
     "stderr": null,
     "stderr_hash": null,
     "returncode": 0

tests/reference/asr-array_01_decl-39cf894.stdout

@@ -6,7 +6,7 @@
     "outfile": null,
     "outfile_hash": null,
     "stdout": "asr-array_02_decl-e8f6874.stdout",
-    "stdout_hash": "a89ba70977945aaa7de41e98302e8fae61815262e2f41e18222a0a80",
+    "stdout_hash": "9fcfaf5cb7d6dfd6798902f102d41743c0c6c89180572345d886154f",
     "stderr": null,
     "stderr_hash": null,
     "returncode": 0

tests/reference/asr-array_02_decl-e8f6874.json

@@ -6,7 +6,7 @@
     "outfile": null,
     "outfile_hash": null,
     "stdout": "asr-bindc_02-bc1a7ea.stdout",
-    "stdout_hash": "48b7a2efdd3ca69877638f9bf332aecea9638fb382e73ede9cce651e",
+    "stdout_hash": "00799f9ef4a45d4c3e93d043b31f73f8038a4b82a4ee191f29b0467f",
     "stderr": null,
     "stderr_hash": null,
     "returncode": 0

tests/reference/asr-array_02_decl-e8f6874.stdout

@@ -6,7 +6,7 @@
     "outfile": null,
     "outfile_hash": null,
     "stdout": "asr-cast-435c233.stdout",
-    "stdout_hash": "0263f1538bbd3457f42a3bed5817bc098151059c3202305815940ac9",
+    "stdout_hash": "17cc5e55a18b44793f2e153ef08fd85e279fb740ad6991a3d47665b8",
     "stderr": null,
     "stderr_hash": null,
     "returncode": 0