diff --git a/libkineto/CMakeLists.txt b/libkineto/CMakeLists.txt index 140ee6882..028b7fc94 100644 --- a/libkineto/CMakeLists.txt +++ b/libkineto/CMakeLists.txt @@ -141,14 +141,18 @@ endif() if (LIBKINETO_NOCUPTI AND LIBKINETO_NOROCTRACER AND LIBKINETO_NOXPUPTI) get_filelist("get_libkineto_cpu_only_srcs(with_api=False)" LIBKINETO_SRCS) message(INFO " CUPTI unavailable or disabled - not building GPU profilers") -elseif(NOT LIBKINETO_NOROCTRACER) - get_filelist("get_libkineto_roctracer_srcs(with_api=False)" LIBKINETO_SRCS) +else() + if(NOT LIBKINETO_NOROCTRACER) + get_filelist("get_libkineto_roctracer_srcs(with_api=False)" LIBKINETO_roc_SRCS) message(INFO " Building with roctracer") -elseif(DEFINED LIBKINETO_NOXPUPTI AND NOT LIBKINETO_NOXPUPTI) - get_filelist("get_libkineto_xpupti_srcs(with_api=False)" LIBKINETO_SRCS) + elseif(NOT LIBKINETO_NOCUPTI) + get_filelist("get_libkineto_cupti_srcs(with_api=False)" LIBKINETO_cuda_SRCS) + endif() + if(DEFINED LIBKINETO_NOXPUPTI AND NOT LIBKINETO_NOXPUPTI) + get_filelist("get_libkineto_xpupti_srcs(with_api=False)" LIBKINETO_xpu_SRCS) message(INFO " Building with xpupti") -else() - get_filelist("get_libkineto_cupti_srcs(with_api=False)" LIBKINETO_SRCS) + endif() + set(LIBKINETO_SRCS ${LIBKINETO_roc_SRCS} ${LIBKINETO_xpu_SRCS} ${LIBKINETO_cuda_SRCS}) endif() get_filelist("get_libkineto_public_headers()" LIBKINETO_PUBLIC_HEADERS) get_filelist("get_libkineto_api_srcs()" LIBKINETO_API_SRCS)