profiler: add PrivateUse1 activity support for custom backends (#50977)
Signed-off-by: David Holtz <56723830+dmholtz@users.noreply.github.com>
David Holtz committed
Aug 11, 2026 at 11:38 UTC
490259c1f63faf025b8050504db63d81c817d781
1 file changed
+2
-1
vllm/profiler/wrapper.py
+2
-1
@@ -150,10 +150,11 @@ class WorkerProfiler(ABC):
150
return nullcontext()
151
152
153
-TorchProfilerActivity = Literal["CPU", "CUDA", "XPU"]
153
+TorchProfilerActivity = Literal["CPU", "CUDA", "PrivateUse1", "XPU"]
154
TorchProfilerActivityMap = {
155
"CPU": torch.profiler.ProfilerActivity.CPU,
156
"CUDA": torch.profiler.ProfilerActivity.CUDA,
157
+ "PrivateUse1": torch.profiler.ProfilerActivity.PrivateUse1,
158
"XPU": torch.profiler.ProfilerActivity.XPU,
159
}
160