illegal instruction

kendryte · Feb 12, 2025 · 08141ae · 08141ae
1 parent c5fdf19
commit 08141ae
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/ntt/test/ctest/test_ntt_binary_add_fp16.cpp b/ntt/test/ctest/test_ntt_binary_add_fp16.cpp
@@ -686,7 +686,7 @@ template <typename T, size_t vl> void test_vector() {
     _TEST_VECTOR(T, 8)
 
 TEST(UnaryTestAdd, vector) {
-    TEST_VECTOR(float)
+    TEST_VECTOR(half)
     TEST_VECTOR(int32_t)
     TEST_VECTOR(int64_t)
 }

diff --git a/ntt/test/ctest/test_ntt_unary_abs_fp16.cpp b/ntt/test/ctest/test_ntt_unary_abs_fp16.cpp
diff --git a/ntt/test/ntt_test.h b/ntt/test/ntt_test.h
@@ -193,8 +193,10 @@ bool compare_tensor(TTensor &lhs, TTensor &rhs, double threshold = 0.999f) {
 
     bool pass = true;
     nncase::ntt::apply(lhs.shape(), [&](auto index) {
-        auto d1 = (double)lhs(index);
-        auto d2 = (double)rhs(index);
+        auto d1 = static_cast<double>(
+            static_cast<typename TTensor::element_type>(lhs(index)));
+        auto d2 = static_cast<double>(
+            static_cast<typename TTensor::element_type>(rhs(index)));
         v1.push_back(d1);
         v2.push_back(d2);
         if (d1 != d2) {