Skip to content

vllm_omni.platforms.npu.profiler

logger module-attribute

logger = init_logger(__name__)

NPUTorchProfilerWrapper

Bases: OmniTorchProfilerWrapper

NPU-specific profiler using torch_npu.profiler.

Key differences from base profiler: - Uses torch_npu.profiler instead of torch.profiler - Different experimental_config options (AiCMetrics, profiler_level, etc.) - Uses tensorboard_trace_handler for trace output - No key_averages() support - requires offline parsing with torch_npu.profiler.profiler.analyse()