Add tests for vset

check_instr.py

@@ -25,6 +25,8 @@
                         encoded.append("vr")
                     elif part[0] == "r":
                         encoded.append("r")
+                    elif part[0] == "c":
+                        encoded.append("fcc")
                     elif part[0] == "u" or part[0] == "s":
                         encoded.append("imm")
                 encoding[name] = ", ".join(encoded)

code/untested.py

@@ -6,17 +6,19 @@
     if os.path.exists(src):
         # make sure we are testing the expected instruction
         if src.split(".")[0] not in open(src, 'r').read():
+            # intrinsic's name is different from instruction's name
+            if src.startswith("vset") or src.startswith("vset"):
+                continue
             print("Bad test:", src)
     if not os.path.exists(src):
         if (
             src.startswith("xvf")
             or src.startswith("vf")
-            or src.startswith("xvset")
-            or src.startswith("vset")
             or src.startswith("xvld")
             or src.startswith("vld")
             or src.startswith("xvst")
             or src.startswith("vst")
+            or src == "common.cpp"
         ):
             # skipped
             continue

code/vsetallnez_b.cpp

@@ -0,0 +1,18 @@
+#include "common.h"
+
+void test() {
+  __m128i a = {0, 0};
+  assert(!__lsx_bnz_b(a));
+
+  a = __m128i {1, 0};
+  assert(!__lsx_bnz_b(a));
+
+  a = __m128i {0, 1};
+  assert(!__lsx_bnz_b(a));
+
+  a = __m128i {0x1111111111111111, 0x1111111111111111};
+  assert(__lsx_bnz_b(a));
+
+  a = __m128i {0x1111111111111100, 0x1111111111111111};
+  assert(!__lsx_bnz_b(a));
+}

code/vsetallnez_d.cpp

@@ -0,0 +1,18 @@
+#include "common.h"
+
+void test() {
+  __m128i a = {0, 0};
+  assert(!__lsx_bnz_d(a));
+
+  a = __m128i {1, 0};
+  assert(!__lsx_bnz_d(a));
+
+  a = __m128i {0, 1};
+  assert(!__lsx_bnz_d(a));
+
+  a = __m128i {0x1111111111111111, 0x1111111111111111};
+  assert(__lsx_bnz_d(a));
+
+  a = __m128i {0x0, 0x1111111111111111};
+  assert(!__lsx_bnz_d(a));
+}

code/vsetallnez_h.cpp

@@ -0,0 +1,18 @@
+#include "common.h"
+
+void test() {
+  __m128i a = {0, 0};
+  assert(!__lsx_bnz_h(a));
+
+  a = __m128i {1, 0};
+  assert(!__lsx_bnz_h(a));
+
+  a = __m128i {0, 1};
+  assert(!__lsx_bnz_h(a));
+
+  a = __m128i {0x1111111111111111, 0x1111111111111111};
+  assert(__lsx_bnz_h(a));
+
+  a = __m128i {0x1111111111110000, 0x1111111111111111};
+  assert(!__lsx_bnz_h(a));
+}

code/vsetallnez_w.cpp

@@ -0,0 +1,18 @@
+#include "common.h"
+
+void test() {
+  __m128i a = {0, 0};
+  assert(!__lsx_bnz_w(a));
+
+  a = __m128i {1, 0};
+  assert(!__lsx_bnz_w(a));
+
+  a = __m128i {0, 1};
+  assert(!__lsx_bnz_w(a));
+
+  a = __m128i {0x1111111111111111, 0x1111111111111111};
+  assert(__lsx_bnz_w(a));
+
+  a = __m128i {0x1111111100000000, 0x1111111111111111};
+  assert(!__lsx_bnz_w(a));
+}

code/vsetanyeqz_b.cpp

@@ -0,0 +1,18 @@
+#include "common.h"
+
+void test() {
+  __m128i a = {0, 0};
+  assert(__lsx_bz_b(a));
+
+  a = __m128i {1, 0};
+  assert(__lsx_bz_b(a));
+
+  a = __m128i {0, 1};
+  assert(__lsx_bz_b(a));
+
+  a = __m128i {0x1111111111111111, 0x1111111111111111};
+  assert(!__lsx_bz_b(a));
+
+  a = __m128i {0x1111111111111100, 0x1111111111111111};
+  assert(__lsx_bz_b(a));
+}

code/vsetanyeqz_d.cpp

@@ -0,0 +1,18 @@
+#include "common.h"
+
+void test() {
+  __m128i a = {0, 0};
+  assert(__lsx_bz_d(a));
+
+  a = __m128i {1, 0};
+  assert(__lsx_bz_d(a));
+
+  a = __m128i {0, 1};
+  assert(__lsx_bz_d(a));
+
+  a = __m128i {0x1111111111111111, 0x1111111111111111};
+  assert(!__lsx_bz_d(a));
+
+  a = __m128i {0x0, 0x1111111111111111};
+  assert(__lsx_bz_d(a));
+}

code/vsetanyeqz_h.cpp

@@ -0,0 +1,18 @@
+#include "common.h"
+
+void test() {
+  __m128i a = {0, 0};
+  assert(__lsx_bz_h(a));
+
+  a = __m128i {1, 0};
+  assert(__lsx_bz_h(a));
+
+  a = __m128i {0, 1};
+  assert(__lsx_bz_h(a));
+
+  a = __m128i {0x1111111111111111, 0x1111111111111111};
+  assert(!__lsx_bz_h(a));
+
+  a = __m128i {0x1111111111110000, 0x1111111111111111};
+  assert(__lsx_bz_h(a));
+}

code/vsetanyeqz_w.cpp

@@ -0,0 +1,18 @@
+#include "common.h"
+
+void test() {
+  __m128i a = {0, 0};
+  assert(__lsx_bz_w(a));
+
+  a = __m128i {1, 0};
+  assert(__lsx_bz_w(a));
+
+  a = __m128i {0, 1};
+  assert(__lsx_bz_b(a));
+
+  a = __m128i {0x1111111111111111, 0x1111111111111111};
+  assert(!__lsx_bz_w(a));
+
+  a = __m128i {0x1111111100000000, 0x1111111111111111};
+  assert(__lsx_bz_w(a));
+}

code/vseteqz_v.cpp

@@ -0,0 +1,12 @@
+#include "common.h"
+
+void test() {
+  __m128i a = {0, 0};
+  assert(__lsx_bz_v(a));
+
+  a = __m128i {1, 0};
+  assert(!__lsx_bz_v(a));
+
+  a = __m128i {0, 1};
+  assert(!__lsx_bz_v(a));
+}

code/vsetnez_v.cpp

@@ -0,0 +1,12 @@
+#include "common.h"
+
+void test() {
+  __m128i a = {0, 0};
+  assert(!__lsx_bnz_v(a));
+
+  a = __m128i {1, 0};
+  assert(__lsx_bnz_v(a));
+
+  a = __m128i {0, 1};
+  assert(__lsx_bnz_v(a));
+}

main.py

@@ -1096,15 +1096,15 @@ def vshuf4i(name):
     def bz_v():
         return instruction(
             intrinsic=f"int __lsx_bz_v (__m128i a)",
-            instr=f"vseteqz.v vr; bcnez",
+            instr=f"vseteqz.v fcc, vr; bcnez",
             desc=f"Expected to be used in branches: branch if the whole vector `a` equals to zero.",
         )
 
     @my_macro(env)
     def bnz_v():
         return instruction(
             intrinsic=f"int __lsx_bnz_v (__m128i a)",
-            instr=f"vsetnez.v vr; bcnez",
+            instr=f"vsetnez.v fcc, vr; bcnez",
             desc=f"Expected to be used in branches: branch if the whole vector `a` is non-zero.",
         )
 
@@ -1113,7 +1113,7 @@ def bz(name):
         width = widths[name]
         return instruction(
             intrinsic=f"int __lsx_bz_{name} (__m128i a)",
-            instr=f"vsetanyeqz.{name} vr; bcnez",
+            instr=f"vsetanyeqz.{name} fcc, vr; bcnez",
             desc=f"Expected to be used in branches: branch if any {width}-bit element in `a` equals to zero.",
         )
 
@@ -1122,7 +1122,7 @@ def bnz(name):
         width = widths[name]
         return instruction(
             intrinsic=f"int __lsx_bnz_{name} (__m128i a)",
-            instr=f"vsetallnez.{name} vr; bcnez",
+            instr=f"vsetallnez.{name} fcc, vr; bcnez",
             desc=f"Expected to be used in branches: branch if all {width}-bit elements in `a` are non-zero.",
         )