Add vstq unit test
Change-Id: Icb259a96803bfa0c07585797dd840da6c3134643
diff --git a/tests/kelvin_isa/BUILD b/tests/kelvin_isa/BUILD
index 5320b6f..1d83e21 100644
--- a/tests/kelvin_isa/BUILD
+++ b/tests/kelvin_isa/BUILD
@@ -233,6 +233,16 @@
)
kelvin_test(
+ name = "vstq",
+ srcs = [
+ "vstq.cc",
+ ],
+ deps = [
+ ":kelvin_test",
+ ],
+)
+
+kelvin_test(
name = "vzip",
srcs = [
"vzip.cc",
diff --git a/tests/kelvin_isa/vstq.cc b/tests/kelvin_isa/vstq.cc
new file mode 100644
index 0000000..35d70d4
--- /dev/null
+++ b/tests/kelvin_isa/vstq.cc
@@ -0,0 +1,113 @@
+/*
+ * Copyright 2023 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "tests/kelvin_isa/kelvin_test.h"
+
+constexpr int kOffset = 2 * 512 / 32;
+
+const uint32_t pattern_[512 / 32] = {
+ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
+};
+
+struct __attribute__((__packed__)) __attribute__((aligned(64))) {
+ uint32_t force_unaligned; // Make data not 64-byte aligned.
+ uint32_t data[4 * kOffset]; // Handle quad * offset
+} buffer_;
+
+uint32_t b_buffer2_[4 * kOffset] __attribute__((aligned(64)));
+
+template <int offset>
+void check(const int i, const uint32_t ref) {
+ const uint32_t dut = buffer_.data[i * offset];
+ if (ref != dut) {
+ printf("**error:vstq_test<%d>(%d) %ld %ld\n", offset, i, ref, dut);
+ exit(-1);
+ }
+}
+
+template <int base, int offset>
+void check_base_shift(const int i, const uint32_t ref) {
+ int vlen;
+ getmaxvl_w(vlen);
+ const uint32_t dut = b_buffer2_[base * vlen / 4 + i * offset * vlen / 4];
+ if (ref != dut) {
+ printf("**error:vstq_test_base_offset<%d, %d>(%d) %ld %ld\n", base, offset,
+ i, ref, dut);
+ exit(-1);
+ }
+}
+
+template <int offset>
+void vstq_test() {
+ uint32_t* ptr = buffer_.data;
+ memset(&buffer_, 0, sizeof(buffer_));
+ vld_w_x(v0, pattern_);
+ vstq_w_sp_xx(v0, ptr, offset);
+
+ check<offset>(0, 1);
+ check<offset>(1, 3);
+ check<offset>(2, 5);
+ check<offset>(3, 7);
+}
+
+template <int base, int offset>
+void vstq_test_base_offset() {
+ int vlen;
+ getmaxvl_w(vlen);
+ uint32_t* ptr = b_buffer2_ + (base * vlen / 4);
+ memset(b_buffer2_, 0, sizeof(b_buffer2_));
+ vld_w_x(v0, pattern_);
+ vstq_w_sp_xx(v0, ptr, offset * vlen / 4);
+
+ check_base_shift<base, offset>(0, 1);
+ check_base_shift<base, offset>(1, 3);
+ check_base_shift<base, offset>(2, 5);
+ check_base_shift<base, offset>(3, 7);
+}
+
+int main() {
+ vstq_test<kOffset + 0>();
+ vstq_test<kOffset - 1>();
+ vstq_test<kOffset + 1>();
+
+ vstq_test_base_offset<0, 1>();
+ vstq_test_base_offset<0, 2>();
+ vstq_test_base_offset<0, 3>();
+ vstq_test_base_offset<0, 4>();
+ vstq_test_base_offset<0, 5>();
+ vstq_test_base_offset<1, 1>();
+ vstq_test_base_offset<1, 2>();
+ vstq_test_base_offset<1, 3>();
+ vstq_test_base_offset<1, 4>();
+ vstq_test_base_offset<1, 5>();
+ vstq_test_base_offset<2, 1>();
+ vstq_test_base_offset<2, 2>();
+ vstq_test_base_offset<2, 3>();
+ vstq_test_base_offset<2, 4>();
+ vstq_test_base_offset<2, 5>();
+ vstq_test_base_offset<3, 1>();
+ vstq_test_base_offset<3, 2>();
+ vstq_test_base_offset<3, 3>();
+ vstq_test_base_offset<3, 4>();
+ vstq_test_base_offset<3, 5>();
+ vstq_test_base_offset<4, 1>();
+ vstq_test_base_offset<4, 2>();
+ vstq_test_base_offset<4, 3>();
+ vstq_test_base_offset<4, 4>();
+ vstq_test_base_offset<4, 5>();
+
+ return 0;
+}
