float32 support

vitalsong · Sep 3, 2022 · 3c46695 · 3c46695
1 parent 5991304
commit 3c46695
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -9,13 +9,21 @@ endif()
 
 option(DSPLIB_POOL_ALLOCATOR "Use pool allocator for vectors" OFF)
 option(DSPLIB_ASAN_ENABLED "Address sanitizer enabled" OFF)
+option(DSPLIB_USE_FLOAT32 "Use float32 for base type dsplib::real_t" OFF)
 
 if (DSPLIB_ASAN_ENABLED)
     include(cmake/sanitizer.cmake)
     enable_address_sanitizer(TARGET dsplib)
 endif()
 
 if (DSPLIB_POOL_ALLOCATOR)
-    message(STATUS "dsplib: use pool allocator")
+    message("dsplib: use pool allocator")
     target_compile_definitions(dsplib PUBLIC "DSPLIB_POOL_ALLOCATOR")
+endif()
+
+if (DSPLIB_USE_FLOAT32)
+    message("dsplib: base type dsplib::real_t = float32")
+    target_compile_definitions(dsplib PUBLIC "DSPLIB_USE_FLOAT32")
+else()
+    message("dsplib: base type dsplib::real_t = float64")
 endif()
diff --git a/include/dsplib/types.h b/include/dsplib/types.h
@@ -41,8 +41,11 @@ namespace dsplib {
 
 //-------------------------------------------------------------------------------------------------
 //base scalar type
+#ifdef DSPLIB_USE_FLOAT32
+using real_t = float;
+#else
 using real_t = double;
-// using real_t = float;
+#endif
 
 constexpr real_t pi = 3.141592653589793238463;