Home
last modified time | relevance | path

Searched refs:max_compile_time_stream_priorities (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/c10/xpu/
H A DXPUStream.cpp28 max_compile_time_stream_priorities>>
31 std::array<std::atomic<uint32_t>, max_compile_time_stream_priorities>>
104 for (const auto p : c10::irange(max_compile_time_stream_priorities)) { in initDeviceStreamState()
221 std::min(-priority, max_compile_time_stream_priorities - 1); in getStreamFromPool()
230 int priority = isHighPriority ? -max_compile_time_stream_priorities + 1 : 0; in getStreamFromPool()
284 for (const auto p : c10::irange(max_compile_time_stream_priorities)) { in syncStreamsOnDevice()
H A DXPUStream.h30 static constexpr int max_compile_time_stream_priorities = 2; variable
135 return std::make_tuple(0, -max_compile_time_stream_priorities + 1); in priority_range()
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAStream.cpp46 c10::cuda::max_compile_time_stream_priorities>
53 c10::cuda::max_compile_time_stream_priorities>
57 stream_flags[c10::cuda::max_compile_time_stream_priorities]
195 max_stream_priorities = range >= c10::cuda::max_compile_time_stream_priorities in initGlobalStreamState()
196 ? c10::cuda::max_compile_time_stream_priorities in initGlobalStreamState()
H A DCUDAStream.h54 static constexpr int max_compile_time_stream_priorities = 4; variable
191 -c10::cuda::max_compile_time_stream_priorities + 1, greatest_priority); in priority_range()