Searched refs:max_compile_time_stream_priorities (Results 1 – 4 of 4) sorted by relevance
28 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()
30 static constexpr int max_compile_time_stream_priorities = 2; variable135 return std::make_tuple(0, -max_compile_time_stream_priorities + 1); in priority_range()
46 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()
54 static constexpr int max_compile_time_stream_priorities = 4; variable191 -c10::cuda::max_compile_time_stream_priorities + 1, greatest_priority); in priority_range()