cpu_rvv.cpp 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. #include <stdio.h>
  2. #if !defined(__riscv) || !defined(__riscv_v)
  3. #error "RISC-V or vector extension(RVV) is not supported by the compiler"
  4. #endif
  5. #if !defined(__THEAD_VERSION__) && defined(__riscv_v_intrinsic) && __riscv_v_intrinsic < 12000
  6. #error "Wrong intrinsics version, v0.12 or higher is required for gcc or clang"
  7. #endif
  8. #include <riscv_vector.h>
  9. #ifdef __THEAD_VERSION__
  10. int test()
  11. {
  12. const float src[] = { 0.0f, 0.0f, 0.0f, 0.0f };
  13. uint64_t ptr[2] = {0x0908060504020100, 0xFFFFFFFF0E0D0C0A};
  14. vuint8m1_t a = vreinterpret_v_u64m1_u8m1(vle64_v_u64m1(ptr, 2));
  15. vfloat32m1_t val = vle32_v_f32m1((const float*)(src), 4);
  16. return (int)vfmv_f_s_f32m1_f32(val);
  17. }
  18. #else
  19. int test()
  20. {
  21. const float src[] = { 0.0f, 0.0f, 0.0f, 0.0f };
  22. uint64_t ptr[2] = {0x0908060504020100, 0xFFFFFFFF0E0D0C0A};
  23. vuint8m1_t a = __riscv_vreinterpret_v_u64m1_u8m1(__riscv_vle64_v_u64m1(ptr, 2));
  24. vfloat32m1_t val = __riscv_vle32_v_f32m1((const float*)(src), 4);
  25. return (int)__riscv_vfmv_f_s_f32m1_f32(val);
  26. }
  27. #endif
  28. int main()
  29. {
  30. printf("%d\n", test());
  31. return 0;
  32. }