fix ndarray_traits<__fp16> test on aarch64

wjakob · Oct 19, 2023 · ab17369 · ab17369
1 parent 6c46b86
commit ab17369
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/test_ndarray.cpp b/tests/test_ndarray.cpp
@@ -15,10 +15,11 @@ static int i_global[] { 1, 2, 3, 4, 5, 6, 7, 8 };
 #if defined(__aarch64__)
 namespace nanobind {
    template <> struct ndarray_traits<__fp16> {
-       static constexpr bool is_float  = true;
-       static constexpr bool is_bool   = false;
-       static constexpr bool is_int    = false;
-       static constexpr bool is_signed = true;
+       static constexpr bool is_complex = false;
+       static constexpr bool is_float   = true;
+       static constexpr bool is_bool    = false;
+       static constexpr bool is_int     = false;
+       static constexpr bool is_signed  = true;
    };
 };
 #endif