Three-Pass Softargmax benchmark (recomputing version)
PiperOrigin-RevId: 275914615
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 0a631cd..f356b60 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -1447,6 +1447,15 @@
TARGET_INCLUDE_DIRECTORIES(f32-spmm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
TARGET_LINK_LIBRARIES(f32-spmm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
+ ADD_EXECUTABLE(f32-softmax-bench bench/f32-softmax.cc)
+ SET_TARGET_PROPERTIES(f32-softmax-bench PROPERTIES
+ CXX_STANDARD 11
+ CXX_STANDARD_REQUIRED YES
+ CXX_EXTENSIONS YES)
+ TARGET_INCLUDE_DIRECTORIES(f32-softmax-bench PRIVATE src)
+ TARGET_INCLUDE_DIRECTORIES(f32-softmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
+ TARGET_LINK_LIBRARIES(f32-softmax-bench PRIVATE XNNPACK fp16 benchmark bench-utils)
+
ADD_EXECUTABLE(q8-gemm-bench bench/q8-gemm.cc)
SET_TARGET_PROPERTIES(q8-gemm-bench PROPERTIES
CXX_STANDARD 11