Searched refs:main_stream (Results 1 – 1 of 1) sorted by relevance
96 se::Stream* main_stream = run_options->stream(); in CheckCompatibilityWithServiceExecutableRunOptions() local99 main_stream->parent()->platform_kind(); in CheckCompatibilityWithServiceExecutableRunOptions()102 main_stream->parent()->GetDeviceDescription().rocm_amdgpu_isa_version( in CheckCompatibilityWithServiceExecutableRunOptions()111 main_stream->parent()->GetDeviceDescription().cuda_compute_capability( in CheckCompatibilityWithServiceExecutableRunOptions()138 se::Stream* main_stream = run_options->stream(); in ExecuteThunks() local139 se::StreamExecutor* executor = main_stream->parent(); in ExecuteThunks()155 sub_streams.back()->ThenWaitFor(main_stream); in ExecuteThunks()158 HloExecutionProfiler profiler(do_profile, hlo_execution_profile, main_stream, in ExecuteThunks()176 (stream_no == 0 ? main_stream : sub_streams[stream_no - 1].get()); in ExecuteThunks()201 auto finish_event = absl::make_unique<se::Event>(main_stream->parent()); in ExecuteThunks()[all …]