OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[XPU][Kineto Submodule] Introduce kineto-based XPU profiler (#130811)

Browse Source 
As XPU became a PyTorch built-in device, the profiler support is indispensable part of functionality completeness. This PR is associated with the PR to introduce XPU profiler plugin into the kineto. When USE_XPU is enabled, the LIBKINETO_NOXPUPTI option will be suppressed accordingly, which allows kineto to build with XPU profiler plugin.

Associated PR to introduce kineto-based XPU profiler into kineto:
https://github.com/pytorch/kineto/pull/961

Also updates the Kineto Submodule to include XPU changes.

Co-authored-by: Aaron Enye Shi <enye.shi@gmail.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/130811
Approved by: https://github.com/aaronenyeshi
This commit is contained in:
Chen, Zejun 2024-08-07 18:41:34 +00:00 committed by PyTorch MergeBot
parent 07551887b8
commit 26b0011fb8
2 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
cmake/Dependencies.cmake
View File

@ -1563,7 +1563,14 @@ if(USE_KINETO)
message(STATUS "Using Kineto with Roctracer support") message(STATUS "Using Kineto with Roctracer support")
endif() endif()
if(LIBKINETO_NOCUPTI AND LIBKINETO_NOROCTRACER) if(NOT USE_XPU)
set(LIBKINETO_NOXPUPTI ON CACHE STRING "" FORCE)
else()
set(LIBKINETO_NOXPUPTI OFF CACHE STRING "")
message(STATUS "Using Kineto with XPUPTI support")
endif()
if(LIBKINETO_NOCUPTI AND LIBKINETO_NOROCTRACER AND LIBKINETO_NOXPUPTI)
message(STATUS "Using CPU-only version of Kineto") message(STATUS "Using CPU-only version of Kineto")
endif() endif()
@ -1662,7 +1669,12 @@ if(USE_KINETO)
if(LIBKINETO_NOROCTRACER) if(LIBKINETO_NOROCTRACER)
string(APPEND CMAKE_CXX_FLAGS " -DLIBKINETO_NOROCTRACER") string(APPEND CMAKE_CXX_FLAGS " -DLIBKINETO_NOROCTRACER")
endif() endif()
if(LIBKINETO_NOCUPTI AND LIBKINETO_NOROCTRACER) if(LIBKINETO_NOXPUPTI)
string(APPEND CMAKE_CXX_FLAGS " -DLIBKINETO_NOXPUPTI=ON")
else()
string(APPEND CMAKE_CXX_FLAGS " -DLIBKINETO_NOXPUPTI=OFF")
endif()
if(LIBKINETO_NOCUPTI AND LIBKINETO_NOROCTRACER AND LIBKINETO_NOXPUPTI)
message(STATUS "Configured Kineto (CPU)") message(STATUS "Configured Kineto (CPU)")
else() else()
message(STATUS "Configured Kineto") message(STATUS "Configured Kineto")

2
third_party/kineto vendored

@ -1 +1 @@
Subproject commit da2f2682cabaf95d601fa2a9b7e0979f84fe7667 Subproject commit 0cded49b0cce22f7d777861db5c937da7f3e22db