Add profiler annotations in common primitives for CUDA backend (#2244)

This commit is contained in:
Cheng
2025-06-05 11:55:12 +09:00
committed by GitHub
parent aede70e81d
commit 52dc8c8cd5
2 changed files with 10 additions and 0 deletions

View File

@@ -17,6 +17,8 @@ target_sources(
${CMAKE_CURRENT_SOURCE_DIR}/utils.cpp
${CMAKE_CURRENT_SOURCE_DIR}/worker.cpp)
target_compile_definitions(mlx PRIVATE MLX_USE_CUDA)
# Enable defining device lambda functions.
target_compile_options(mlx
PRIVATE "$<$<COMPILE_LANGUAGE:CUDA>:--extended-lambda>")