Enable fp16 for risc-v

PiperOrigin-RevId: 634141528
google · May 16, 2024 · 30080c2 · 30080c2
1 parent 8d5f4e0
commit 30080c2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/configs/hardware-config.c b/src/configs/hardware-config.c
@@ -160,6 +160,11 @@ static void init_hardware_config(void) {
       __asm__(".word 0xC22022F3"  /* CSRR t0, vlenb */ : "=r" (vlenb));
       hardware_config.vlenb = vlenb;
       xnn_log_info("RISC-V VLENB: %" PRIu32, vlenb);
+
+      // Assume a uarch with long vectors supports fp16 as well
+      if (vlenb >= 64) {
+        hardware_config.use_riscv_vector_fp16_arith = true;
+      }
     }
   #endif